Jetson Orin系列主板 - Pytorch不兼容问题
可能出现的问题一:GPU无法使用
安装jetson版本的pytorch
教程:https://docs.nvidia.com/deeplearning/frameworks/install-pytorch-jetson-platform/index.html
可以自己选用版本:https://developer.download.nvidia.cn/compute/redist/jp/
我使用的版本:
torch-2.5.0a0+872d972e41.nv24.08.17622132-cp310-cp310-linux_aarch64.whl
下载后,运行:
export TORCH_INSTALL=torch-2.5.0a0+872d972e41.nv24.08.17622132-cp310-cp310-linux_aarch64.whl
pip install --no-cache $TORCH_INSTALL
验证:
import torch
torch.cuda.is_available()
可能出现的问题二:
报错 ImportError: libcusparseLt.so.0: cannot open shared object file: No such file or directory
解决方法:
安装 Jetson Orin 平台兼容的 CUDA 版本对应的 cuSPARSELt 库
下载链接:
https://developer.nvidia.com/cuda-12-6-0-download-archive?target_os=Linux&target_arch=aarch64-jetson&Compilation=Native&Distribution=Ubuntu&target_version=22.04&target_type=deb_local
https://developer.nvidia.com/cusparselt-downloads
可能出现的问题三:没有torchvision
解决方法:
手动安装匹配的vision版本,torch 2.5->torchvision 0.20.0
git clone --branch v0.20.0 https://github.com/pytorch/vision.git
下载后编译:
export BUILD_VERSION=0.20.0
python3 setup.py install --user
更多推荐




所有评论(0)