当前位置: 代码迷 >> 综合 >> Jetson TX2-JetPack 4.3 安装 pytorch和torchvision
  详细解决方案

Jetson TX2-JetPack 4.3 安装 pytorch和torchvision

热度:19   发布时间:2023-12-13 23:26:38.0

声明:

TX2系统: Ubuntu18.04
python版本:3.6.9

刷入JetPack 4.3apt更换为国内源。选择这个版本的原因是对应cuda的版本为10.0。(JetPack 4.4对应cuda版本为10.2,如果需要cuda 10.2版本,请刷入4.4。)


本文基于JetPack4.3,安装pytorch版本为1.2torchvision版本为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