目录
首先,torch和torchvision都不能直接pip安装,
以下的演示是torch1.10.0 + torchvision v0.11.1
1、安装torch
torch要去英伟达官网上找。链接如下(科学上网)https://forums.developer.nvidia.com/t/pytorch-for-jetson/72048,(百度网盘里也有,后边在写)
上边网址打开之后是这样
根据你的cuda版本(查询cuda版本,命令用nvcc -v),我是cuda10.2,按如下对应关系,
下载torch版本,挑一个下载就行,下载完的文件像这样,这三个文件的百度网盘链接在最后
下载完后用u盘转到jetson nx 板子上,主目录下新建个文件夹放好(位置没有要求,这样做就是为了删除的时候好找)
在板子上,进入放torch的文件夹,打开终端,进入你的conda虚机环境(如果你有的话)
运行
pip3 install torch-1.10.0-cp36-cp36m-linux_aarch64.whl
注意torch文件名换成自己的,安装完成后,会提示successful install torch1.10.0,就安装成功了