博客

（1）主要参考

（2）官网下载地址

https://forums.developer.nvidia.com/t/pytorch-for-jetson-version-1-8-0-now-available/72048

（3）注意自己的镜像对应的版本，这里推荐下载1.8版本

PyTorch v1.8.0

• JetPack 4.4 (L4T R32.4.3) / JetPack 4.4.1 (L4T R32.4.4) / JetPack 4.5 (L4T R32.5.0) / JetPack 4.5.1 (L4T R32.5.1)
  • Python 3.6 –
    
    
torch-1.8.0-cp36-cp36m-linux_aarch64.whl

    3.4k
    

如果无法下载，进入如下好人的链接

https://download.csdn.net/download/qq_51095731/16753369

（3）安装方法

sudo apt-get install python3-pip libopenblas-base libopenmpi-dev 
pip3 install Cython
pip3 install numpy 
pip3 install pip3 install torch-1.8.0-cp36-cp36m-linux_aarch64.whl 

(4)ceshixia

python3
import torch
print(torch.cuda.is_available())

（5）

安装 torchvision

说明地址：https://github.com/pytorch/vision

因为我安装的是1.8.0的pytorch，对应

torchvision

版本是0.9.0，所以安装如下：、

sudo apt-get install libjpeg-dev zlib1g-dev libpython3-dev libavcodec-dev libavformat-dev libswscale-dev

#shangmian de  xian install
pip3 install torchvision==0.9.0  #我用这条没有安装成功

because pip3 install torchvision==0.9.0  #我用这条没有安装成功,  手动去这个地址下载代码包：

https://github.com/pytorch/vision/tree/v0.9.0

，手动选择下载

解压后，进入文件夹

export BUILD_VERSION=0.9.0
sudo python3 setup.py install 

一定要耐心等待，如果等不了，可以通过jtop看看情况

jtop

中间用到pillow>=4.1.1如果比较慢建议先安装

pip3 install -i https://pypi.doubanio.com/simple/ --trusted-host pypi.doubanio.com pillow

#测试是否安装成功
python3
import torchvision
print(torchvision.__version__)