TX2 Jetpack4.3安装pytorch1.4.0+torchvision0.5.0

标题TX2 Jetpack4.3安装pytorch1.4.0+torchvision0.5.0

1. 查看安装的jetpack版本

sudo pip3 install jetson-stats
sudo jtop

然后选择5-INFO（键盘按5） 就可以清楚看到 Jetpack 版本，还有其他的硬件信息；

2. 根据jetpack版本查看支持的pytorch版本

由于TX2是arm架构，所以在pytorch官网无法安装，目前清华源好像有1.8.0和1.8.1版本的pytorch（感谢苹果M1带动arm行业发展？）
首先可以去pytorch for jetson 官网查看所需版本，如下图：

选择下载需要的版本，我的是jetpack4.3版本，选择了Python 3.6 - torch-1.4.0-cp36-cp36m-linux_aarch64.whl
由于不可描述的原因，下载很慢，直接失败，下面推荐百度云下载[链接]提取码：5j45 (https://pan.baidu.com/s/11oSQ1n-mOGAx8BOVDiEx9w) ，有条件的直接在官网下载新的，我之前也是下载别人的1.4.0安装之后报错 “非法指令（核心已转储）”表示版本不对应；
下载之后直接pip安装即可。

3. 安装torchvision

$ sudo apt-get install libjpeg-dev zlib1g-dev libpython3-dev libavcodec-dev libavformat-dev libswscale-dev
$ git clone --branch v0.5.0 https://hub.fastgit.org/pytorch/vision torchvision   # see below for version of torchvision to download
$ cd torchvision
$ export BUILD_VERSION=0.5.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

先修改对应的版本，具体对应关系见下图：

4. 测试

>>> import torch
>>> print(torch.__version__)
>>> print('CUDA available: ' + str(torch.cuda.is_available()))
>>> print('cuDNN version: ' + str(torch.backends.cudnn.version()))
>>> a = torch.cuda.FloatTensor(2).zero_()
>>> print('Tensor a = ' + str(a))
>>> b = torch.randn(2).cuda()
>>> print('Tensor b = ' + str(b))
>>> c = a + b
>>> print('Tensor c = ' + str(c))

>>> import torchvision
>>> print(torchvision.__version__)

没有报错安装成功了！
下面是我安装成功后的 pip list 各包的版本

appdirs             1.4.4
certifi             2020.12.5
chardet             4.0.0
cmake               3.18.4.post1
Cython              0.29.23
distlib             0.3.1
filelock            3.0.12
idna                2.10
importlib-metadata  3.7.3
importlib-resources 5.1.2
numpy               1.19.3
opencv-python       3.4.13.47
pbr                 5.5.1
Pillow              6.2.2
pip                 21.0.1
requests            2.25.1
setuptools          49.6.0.post20210108
stevedore           3.3.0
torch               1.4.0
torchvision         0.5.0
typing-extensions   3.7.4.3
urllib3             1.26.4
virtualenv          20.4.3
virtualenv-clone    0.5.4
virtualenvwrapper   4.8.4
wheel               0.36.2
zipp                3.4.1

5. 关于报错

>>> import torch
非法指令 (核心已转储)

安装了不合适版本的torch,同样的安装不合适版本的numpy也会报错；

另外，我再编译torchvision的时候最后停下了，大概是这个样子：

Installed /home/nvidia/.local/lib/python3.6/site-packages/torchvision-0.5.0-py3.6-linux-aarch64.egg
Processing dependencies for torchvision==0.5.0
Searching for pillow>=4.1.1
Reading https://pypi.org/simple/pillow/
Downloading https://files.pythonhosted.org/packages/84/96/16ae48ded4a14a47bfceba39627625a91f1fb66e6d2d2214d5317f105cce/Pillow-8.2.0-cp36-cp36m-manylinux2014_aarch64.whl#sha256=c5236606e8570542ed424849f7852a0ff0bce2c4c8d0ba05cc202a5a9c97dee9

应该是安装pillow库的网络有问题，重新执行命令即可！