三、 Jetson Xavier NX上Pytorch环境安装

官网安装教程：PyTorch for Jetson - version 1.11 now available - Jetson & Embedded Systems / Jetson Nano - NVIDIA Developer Forums

基本按照教程来，就没什么问题。

Pytorch安装

wget https://nvidia.box.com/shared/static/p57jwntv436lfrd78inwl7iml6p13fzh.whl -O torch-1.8.0-cp36-cp36m-linux_aarch64.whl
sudo apt-get install python3-pip libopenblas-base libopenmpi-dev libomp-dev
pip3 install Cython
pip3 install numpy torch-1.8.0-cp36-cp36m-linux_aarch64.whl

在安装时，长期得不到响应，估计是因为外网的原因，得挂梯子才行。可以先将该文件下载，进行本地安装。直接运行命令

pip3 install numpy torch-1.8.0-cp36-cp36m-linux_aarch64.whl

等待一会儿，安装完成：

简单的检验下安装是否成功：

Torchvision安装

官方的教程：

$ sudo apt-get install libjpeg-dev zlib1g-dev libpython3-dev libavcodec-dev libavformat-dev libswscale-dev
$ git clone --branch <version> https://github.com/pytorch/vision torchvision   # see below for version of torchvision to download
$ cd torchvision
$ export BUILD_VERSION=0.x.0  # where 0.x.0 is the torchvision version  
$ python3 setup.py install --user
$ cd ../  # attempting to load torchvision from build dir will result in import error
$ pip install 'pillow<7' # always needed for Python 2.7, not needed torchvision v0.5.0+ with Python 3.6

我呢，对照官方的教程，对应的改下版本号就可以了，官方给出了不同版本Pytorch与对应的Torchvision版本号。

sudo apt-get install libjpeg-dev zlib1g-dev libpython3-dev libavcodec-dev libavformat-dev libswscale-dev
git clone --branch v0.9.0 https://github.com/pytorch/vision torchvision  # 我的pytorch对应的是0.9.0的torchvision
cd torchvision
export BUILD_VERSION=0.9.0
python3 setup.py install --user

耗时挺久的，安装完成：

简单的检验下安装是否成功：

在import torchvision后，出现过AttributeError: module 'PIL.Image' has no attribute 'BILINEAR’的问题，是由于pillow版本的问题，先卸载当前的版本，安装一个较低版本即可。