NX配置pytorch，torchvision环境

最新推荐文章于 2025-09-20 10:38:59 发布

原创最新推荐文章于 2025-09-20 10:38:59 发布 · 356 阅读

1 ·

CC 4.0 BY-SA版权

Jetson NX 学习专栏收录该内容

1 篇文章

订阅专栏

该博客介绍了如何在基于Ubuntu18.04的NX系统上安装Python3.6、CUDA10.2和JetPack4.4环境下，手动编译PyTorch1.6.0和TorchVision0.7.0。首先，通过pip3安装PyTorch的whl文件，然后克隆torchvision源码并编译安装。安装过程中需要注意PyTorch和TorchVision的版本对应，并确保正确使用sudo权限。

1. 系统版本情况

NX系统：Ubuntu18.04
python版本：3.6.9
JetPack：4.4
cuda版本：10.2
安装的Pytorch版本为1.6.0，torch vision版本为0.7.0.

2. 安装pytorch

推荐网上下载whl安装包安装，安装包下载地址在这里，因为NX是arm架构的，所以这里下载aarch64的安装包，如：Python 3.6 - torch-1.6.0-cp36-cp36m-linux_aarch64.whl下载完成后，执行

pip3 install XXX.whl

XXX.whl为下载的安装包名称。
安装完成后，执行一下指令测试：

python3
import torch

如果不报错，则安装成功。

3. 安装torchvision

安装好pytorch后，选择合适版本的torchvision安装，pytorch和torchvision有严格的对应关系，这里要注意，这里给出各个版本的对应关系。
因为torchvision不像torch那样有安装包可以下载下来本地安装，找不到torchvision aarch64架构的安装包，所以这里需要下载源码手动编译安装才行。
torchvision源码地址。
安装指令：

git clone --branch v0.7.0 https://github.com/pytorch/vision torchvision 
cd torchvision
export BUILD_VERSION=0.7.0 
sudo python setup.py install

这里一定记得加sudo。
如果cuda版本正确的话，一般编译不会有错误。编译完成后，在python3中import测试。不报错则安装成功。

有问题可以留言哦。