NVIDIA Jetson Orin NX (Ubuntu 22.04, JetPack 6.2, CUDA 12 cuDnn9) 上成功安装torch2和torchvision
本文介绍了在NVIDIA Jetson Orin NX上安装PyTorch 2.5.0和torchvision 0.20的三种方案,其中推荐使用预编译的whl文件安装。同时提供了cuSPARSELt库的安装方法,并解决了因NumPy版本不兼容导致的错误问题。文章还指出当前JetPack 6.2尚未有官方适配的PyTorch 2版本,建议等待官方更新。通过详细的安装步骤和错误处理方法,帮助用户在Je
一. 我的NVIDIA Jetson Orin NX环境:

二.安装 torch 2.5.0 和 torchvision 0.20
第一种方案:(推荐)
pip install https://github.com/ultralytics/assets/releases/download/v0.0.0/torch-2.5.0a0+872d972e41.nv24.08-cp310-cp310-linux_aarch64.whl
pip install https://github.com/ultralytics/assets/releases/download/v0.0.0/torchvision-0.20.0a0+afc54f7-cp310-cp310-linux_aarch64.whl
第二种方案:
也可以用下面的地址进行安装torch,但这里没有找到torchvision:
pip install torch==2.5.0a0+872d972e41.nv24.08 -f https://developer.download.nvidia.cn/compute/redist/jp/v61/pytorch/
第三种方案:(暂不推荐)
NVIDIA官方PyTorch for Jetson页面是nvidia官方编译好的torch2的版本,但截止2025.10.28, 暂时没有找到适配 JetPack 6.2的版本,可能以后官方会出适配版本。如果强制安装torch2.3版本,会出现找不到cuDnn8的库的错误问题,导致无法正常运行torch。
三.安装 cuSPARSELt 的依赖性问题
下载cuSPARSELt ,里面有安装教程,安装方式可能会随着版本不同而变化,所以要按照官方的安装教程来。
wget https://developer.download.nvidia.com/compute/cusparselt/0.8.1/local_installers/cusparselt-local-tegra-repo-ubuntu2204-0.8.1_0.8.1-1_arm64.deb
sudo dpkg -i cusparselt-local-tegra-repo-ubuntu2204-0.8.1_0.8.1-1_arm64.deb
sudo cp /var/cusparselt-local-tegra-repo-ubuntu2204-0.8.1/cusparselt-*-keyring.gpg /usr/share/keyrings/
sudo apt-get update
sudo apt-get -y install cusparselt
四.解决其他错误问题
import torch出现下面的错误:
A module that was compiled using NumPy 1.x cannot be run in
NumPy 2.2.6 as it may crash. To support both 1.x and 2.x
versions of NumPy, modules must be compiled with NumPy 2.0.
Some module may need to rebuild instead e.g. with 'pybind11>=2.12'.
If you are a user of the module, the easiest solution will be to
downgrade to 'numpy<2' or try to upgrade the affected module.
We expect that some modules will need time to support NumPy 2.
...
降低numpy版本即可:
pip install "numpy<2"
最后再自己导入torch和torchvision包验证一下即可。
参考文章:
https://blog.csdn.net/weixin_53499643/article/details/150064167
https://blog.csdn.net/ydscc/article/details/150209785
鲲鹏昇腾开发者社区是面向全社会开放的“联接全球计算开发者,聚合华为+生态”的社区,内容涵盖鲲鹏、昇腾资源,帮助开发者快速获取所需的知识、经验、软件、工具、算力,支撑开发者易学、好用、成功,成为核心开发者。
更多推荐



所有评论(0)