NVIDIA AGX Xavier 安装torch和对应torchvision

NVIDIA AGX Xavier 安装torch和torchvision

一、pytorch安装

官方网站NVIDIA Jetson Zoo提供了pytorch的arrch64编译版本及安装教程。
首先结合python和JetPack版本从Jetson Zoo下载相应whl文件。我的JetPack是4.5.1，因此下载了pytorch 1.6.0。（从jetson zoo下载似乎需要梯子）

下载完成后，在whl文件处打开终端，并执行以下指令安装pytorch。安装torch的同时会自动安装numpy和future。

sudo apt-get install libopenblas-base libopenmpi-dev python3-pip
pip3 install Cython
pip3 install numpy
torch-1.6.0-cp36-cp36m-linux_aarch64.whl
安装完成后，在虚拟环境中执行下列指令以确保python能正确调用pytorch。

python //进入python
import torch torch.version
//若安装成功且能正常调用，此处能输出安装的pytorch版本 quit() //退出python

我在初次执行以上代码时报错：Illegal instruction (core dumped)。查阅资料发现若安装的是numpy 1.19.5则会出现以上错误，而安装pytorch时自动安装的numpy即1.19.5。Github上讨论给出了一些不同解决方法，我选择了比较简单的一种：卸载numpy 1.19.5，重新安装numpy1.19.4即可解决。

conda remove numpy
pip install numpy==1.19.4

二、torchvision 安装

根据torch版本选择相应版本的torchvision下载安装即可。下图给出部分对应关系，详细对应关系见pypi网站。

我的pytorch是1.6.0，因此安装torchvision0.7.0。

git clone -b v0.7.0 https://github.com/pytorch/vision
cd vision
sudo python3 setup.py install

我在初次执行以上指令时报错：fatal error: libavcodec/avcodec.h: No such file or directory。参考博主 点亮~黑夜 的文章[2]，通过修改setup.py文件，注释掉if has ffmpeg 改为if False，从而解决问题。

安装完成后，在虚拟环境中执行下列指令以确保python能正确调用torchvisin。

python //进入python
import torchvision
torchvision.version //若安装成功且能正常调用，此处能输出安装的torchvision版本
quit()

三、torch和torchvision版本对应关系

---

欢迎大家扫码关注本人公众号：编程复盘与思考随笔

(关注后可以免费获得本人在csdn发布的资源源码)

公众号主要记录编程和刷题时的总结复盘笔记和心得！并且分享读书、工作、生活中的一些思考感悟！

想要组队一起参加阿里天池，kaggle，百度飞浆，科大讯飞等AI相关的比赛的同学可以扫下面的二维码加微信一起讨论学习！