[Jetson Orin Nano] 젯슨 오린 나노 PyTorch 세팅 (JetPack 6.2 전용)

 

1. 필수 시스템 패키지 및 가상 환경 툴 설치

sudo apt-get update
sudo apt-get install -y python3-pip libopenblas-dev virtualenv

 

2. 가상 환경 생성 및 진입

python3 -m virtualenv -p python3 pt_env
source pt_env/bin/activate

 

 

입력창 맨 앞에 (pt_env)가 떠 있으면 성공

 

3. 호환되는 NumPy 버전 세팅 (충돌 방지)

pip3 install "numpy<2.0.0"

 

 

4. PyTorch 본체 설치

pip3 install torch torchvision torchaudio --index-url https://pypi.jetson-ai-lab.io/jp6/cu126 --trusted-host pypi.jetson-ai-lab.io

 

5. 빠진 CUDA 라이브러리(부품) 단독 추가 설치

pip3 install nvidia-cudss-cu12 --no-deps

 

6. 라이브러리 경로 영구 지정

echo 'export LD_LIBRARY_PATH=/home/[PC 이름 집어넣기]/pt_env/lib/python3.10/site-packages/nvidia/cu12/lib:$LD_LIBRARY_PATH' >> ~/pt_env/bin/activate

중간에 대괄호 부분에는 각자의 터미널에서 뜨는 이름을 집어넣으면 된다.

 

7. 가상 환경 새로고침 및 최종 테스트

deactivate
source pt_env/bin/activate
python3 -c "import torch; print('CUDA 사용 가능 여부:', torch.cuda.is_available())"

 

 

True가 뜨면 성공