在NVIDIA Jetson Orin NX (Ubuntu 22.04, JetPack 5.1, CUDA 11 cuDnn8) 上安装PyTorch 2

NVIDIA Jetson Orin NX安装PyTorch 2(GPU版)

我的环境

Ubuntu 22.04, JetPack 5.1, CUDA 11 cuDnn8

---

安装Miniforge的conda创建虚拟环境

参考我的另一篇文章jetson安装Miniforge的Conda和Anaconda3并存切换使用

---

Jetson开发板信息和torch版本对应一致

下面两张图中, 一定要把蓝色箭头的信息和torch版本完全对应上, 不然很有可能出现各种问题.
jetson开发板执行sudo jtop查看信息:

下载与上面版本一致的torch, PyTorch for Jetson 下载到本地, 在本地安装

本地执行pip安装即可.

pip install torch-2.1.0a0+41361538.nv23.06-cp38-cp38-linux_aarch64.whl