Jetson Orin에서 PyTorch 비호환 문제
발생할 수 있는 문제 1: GPU를 사용할 수 없음
jetson 버전 pytorch를 설치합니다 튜토리얼: https://docs.nvidia.com/deeplearning/frameworks/install-pytorch-jetson-platform/index.html
버전은 직접 선택 가능: https://developer.download.nvidia.com/compute/redist/jp/
사용한 버전: torch-2.5.0a0+872d972e41.nv24.08.17622132-cp310-cp310-linux_aarch64.whl
다운로드 후 실행:
Plain
export TORCH_INSTALL=torch-2.5.0a0+872d972e41.nv24.08.17622132-cp310-cp310-linux_aarch64.whl
pip install --no-cache $TORCH_INSTALL테스트:
Python
import torch
torch.cuda.is_available()발생할 수 있는 문제 2:
ImportError: libcusparseLt.so.0: cannot open shared object file: No such file or directory
해결 방법:
Jetson Orin 플랫폼과 호환되는 CUDA 버전에 해당하는 cuSPARSELt 라이브러리를 설치합니다
다운로드 링크:
https://developer.nvidia.com/cusparselt-downloads
발생할 수 있는 문제 3:
torchvision is not available
해결 방법:
호환되는 vision 버전을 수동으로 설치합니다. torch 2.5 → torchvision 0.20.0
Python
git clone --branch v0.20.0 [https://github.com/pytorch/vision.git](https://link.zhihu.com/?target=https%3A//github.com/pytorch/vision.git)다운로드 후 컴파일:
Plain
export BUILD_VERSION=0.20.0
python3 setup.py install --user