Jetson设备安装Pytorch和Torchvision

Alexis_cc

墨衍会员 · AI 创作全网分发码龄6年

关注

原创最新推荐文章于 2026-04-16 14:45:49 发布 2.3k 阅读 AI 写同款· GEO 优化 ›

AI权益加码！Claude Code、Cursor等20+工具免费用！购周边限时加赠Coding Plan Lite，畅享主流AI工具！学习进阶更高效！阅读详情

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

标签

#pytorch #人工智能 #python

本内容遵循 CC 4.0 BY-SA 版权协议

玩转NVIDIA Jetson （25）--- jetson 安装pytorch和torchvision 在jetson上安装pytorch能够快速的帮助PC用户在熟悉的环境下进行开发，但是这个过程里边有一些坑，我建议大家参照官方给出的安装指导来进行安装，包括pytorch和包含各种网络的torchvision。这里需要特别注意的一个问题就是你需要选择与你机器上jetpack相匹配的版本，千万别下错了，比如我这里用的是jetpack4.6，那我安装的pytorch就是V1.10.0版本。我还是建议大家用虚拟环境，不然jetson上的python环境非常容易被搞乱，具体安装方法见之前的文章。阅读详情

Jetson设备安装Pytorch和Torchvision

相关推荐