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安装即可.
bash
pip install torch-2.1.0a0+41361538.nv23.06-cp38-cp38-linux_aarch64.whl