[NVIDIA Jetson] Installing PyTorch for Jetson Platform
·
Robotics/Jetson
Jetson 환경에서 pip install torch 명령어를 사용하면 PyTorch는 설치되지만, torch.cuda.is_available()은 False를 반환하고 GPU를 사용할 수 없다. 이는 PyPI에 등록된 PyTorch가 Jetson(ARM 아키텍처, aarch64) 에서 실행되도록 빌드되지 않았기 때문이다.Jetson에서는 CUDA 및 cuDNN이 포함된 NVIDIA JetPack SDK 환경에 최적화된 별도 PyTorch 빌드가 필요하다. 이를 해결하려면 NVIDIA에서 제공하는 공식 wheel 파일을 설치해야 한다. 1. 현재 JetPack 버전 확인Jetson에 접속하여 다음 명령어로 현재 설치된 JetPack 버전을 확인한다:sudo apt-cache show nvidia-jetp..