Jetson设备安装Pytorch和Torchvision

1.概述

Jetson平台上的Pytorch

  Pytorch(适用于JetPack）是一个针对深度学习优化的张量库，能够调用 GPU 和 CPU。自动微分是通过基于磁带的系统在功能层和神经网络层级别完成的。此功能带来了高水平的灵活性，且具备深度学习框架的速度，并提供了类似 NumPy 的加速功能。这些 NVIDIA 提供的可再发行组件Python pip wheel 安装程序是适用于 PyTorch 的 ，且具有 GPU 加速和对 cuDNN 的支持。这些软件包需要安装在指定的 JetPack 版本。

2.对应版本列表

Torchvision对应Pytorch版本列表 

 

3.安装说明

Pytorch安装方式：（以2.1版本为例）

sudo apt-get install python3-pip libopenblas-base libopenmpi-dev libomp-dev
pip3 install 'Cython<3'
pip3 install numpy torch-2.1.0a0+41361538.nv23.06-cp38-cp38-linux_aarch64.whl

Torchvision安装方式：（根据上表可知，Pytorch版本为2.1，则对应的Torchvision的版本则为0.16.1） 

sudo apt-get install libjpeg-dev zlib1g-dev libpython3-dev libopenblas-dev libavcodec-dev libavformat-dev libswscale-dev
git clone --branch v0.16.1 https://github.com/pytorch/vision torchvision   
cd torchvision
export BUILD_VERSION=0.16.1 
python3 setup.py install --user

验证方式：

Pytorch验证：

>>> 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))

Torchvision验证：

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

为了方便大家下载whl文件，我已全部下载放在阿里云盘，大家有需要的可以自取。

https://www.alipan.com/s/KaDBAayYa9P

提取码: o5o2