首先
先下载arm架构下得torch及torchvision文件夹
PyTorch for Jetson - Announcements - NVIDIA Developer Forums
根据自己所需要的版本下载,点一下就会自动下载
然后上传到jetson Orin nano开发板上,上传软件哪个都行
此网站:torchvision得githup上的
GitHub - pytorch/vision: Datasets, Transforms and Models specific to Computer Vision
命令:下载torchvision
git clone -b v0.8.1 https://github.com/pytorch/vision torchvision
完事之后开始安装
先安装torch
pip install (下载的torch整条名)
安装成功后命令输入看一下有么有
命令:
pip show torch
然后开始安装torchvision,先进入此文件夹
cd torchvision
python3 setup.py install --user
然后开始安装,比较慢 等彻底结束
安装完之后就是网上找个代码,验证一下cuda|torch有没有问题了,输出True,那就基本没啥大问题了
出现bug
这里报错是因为没有依赖,直接下载就可以了
sudo apt-get install libopenblas-base