Jetson Orin开发板，在conda环境中直接实现Pytorch的GPU加速

zlkingdom2026-06-04 14:14

如果通过传统的pip方式安装，则必装出CPU的版本，如果想直接使用系统的默认配置的Pytorch+GPU方案，只需要两个步骤即可

第一，卸载已经在环境下安装的包

复制代码

conda activate 你的环境名
pip uninstall torch torchvision -y

第二，指向系统原生的设置

bash 复制代码

echo "import sys; sys.path.append('/usr/local/lib/python3.10/dist-packages')" > ~/anaconda3/envs/你的环境名/lib/python3.10/site-packages/system_torch.pth

记得在执行命令时替换自己的环境名

然后在环境中再进行一次调用的验证

python 复制代码

python -c "import torch; print(torch.__version__, torch.cuda.is_available())"

看到命令返回True即为成功