一. 我的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

Logo

鲲鹏昇腾开发者社区是面向全社会开放的“联接全球计算开发者,聚合华为+生态”的社区,内容涵盖鲲鹏、昇腾资源,帮助开发者快速获取所需的知识、经验、软件、工具、算力,支撑开发者易学、好用、成功,成为核心开发者。

更多推荐