声明:
TX2系统: Ubuntu18.04
python版本:3.6.9
刷入JetPack 4.3
,apt
更换为国内源。选择这个版本的原因是对应cuda
的版本为10.0
。(JetPack 4.4
对应cuda
版本为10.2
,如果需要cuda 10.2
版本,请刷入4.4
。)
本文基于JetPack4.3
,安装pytorch
版本为1.2
,torchvision
版本为0.4.0
1、安装pytorch 1.2
pytorch1.2.0 CSDN下载链接,下载完之后运行:
pip3 install torch-1.3.0-cp36-cp36m-linux_aarch64.whl
2、安装torchvision 0.4.0
首先安装libjpeg-dev zlib1g-dev
sudo apt install libjpeg-dev zlib1g-dev
然后安装pillow
,注意顺序
pip3 install 'pillow<7'
最后安装torchvision
,版本一定要是0.4.0
git clone --branch v0.4.0 https://github.com/pytorch/vision torchvision
cd torchvision
export BUILD_VERSION=0.4.0
sudo python setup.py install
完