시작이 반이라면
환경 세팅은 60프로 정도인 것 같다
Jetson orin nano Jetpack 6.2 기준 pytorch 설치에 관해 겪었던 어려움들을 적어본다
1. 그래서 내 Cuda 버전이 몇이라고?
엔비디아 Gpu 가속에 Cuda는 원수이면서 친구이다
다행히 젯슨 오린 나노의 os 제트팩에는 쿠다 버전이 명시되어 있다
쿠다 12.6
이것에 맞는 파이토치 관련 패키지 설치를 해보자
이전에
『
python -m pip install --upgrade pip setuptools wheel && \
python -m pip uninstall -y torch torchvision torchaudio && \
python -m pip install --no-deps --force-reinstall \
--index-url https://pypi.jetson-ai-lab.dev/jp6/cu126 \
torch==2.7.0 \
torchvision==0.22.0 \
torchaudio==2.7.0 \
"numpy<2" 』 에잇 귀찮아라며 이걸 복붙하니 문제가 생겼다
서버가 먹통이란다
결국 미리 다운받아둔 whl 파일을 사용하기로 했다
torch-2.8.0-cp310-cp310-linux_aarch64.whl
torchaudio-2.8.0-cp310-cp310-linux_aarch64.whl
torchvision-0.23.0-cp310-cp310-linux_aarch64.whl
이걸 웹에서 찾아서 설치하자
정상적으로 설치가 완료되었다
결론 : 저 whl 파일을 3개 찾아서 설치하면 큰 문제 없이 gpu 가속이 가능한 것을 볼 수 있다