Jetson AGX Xavier安装torch、torchvision

环境ubuntu20.4

1、安装pycharm

2、在pycharm中安装需要的各个插件

3、安装离线torch，下载安装包

Jetson Zoo - eLinux.org

在命令行输入jtop，查看jetpack版本，注意下载对应版本的torch

 4、安装torch

进入pycharm新建环境的目录，把.whl放在这个目录下，执行pip install  torch-1.11.0-cp38-cp38-linux_aarch64.whl

torch只能离线安装

torch下载地址;Jetson Zoo - eLinux.org

或https://forums.developer.nvidia.com/t/pytorch-for-jetson/72048

对应关系：

https://blog.csdn.net/shiwanghualuo/article/details/122860521

5、下载并安装torchvision

git clone -b v0.12.0 https://github.com/pytorch/vision.git vision-0.12.0

cd vision-0.12.0
sudo /home/jetson/archiconda/envs/yolov5env/bin/python3.6 setup.py install

6、避免踩坑

Jetson AGX Xavier不支持nvidia-smi命令，使用nvcc -V