라벨이 torchvision인 게시물 표시

[Jetson] Jetson Nano Pytorch 1.11.0, Torchvision 0.12.0 설치 방법

이미지
Jetson Nano Pytorch 1.11.0 & Torchvision 0.12.0 설치 방법 Jetson Nano에서 YOLOv8을 동작하기 위해선 python3.8 이상의 버전이 필요합니다. Nano에 호환되는 JetPack 4.6.1에는 python3.6이 설치되었기 때문에 여기 를 참고해서 3.8를 설치합니다. 설치를 진행하기 앞서 venv를 활성화했다고 가정하고 진행하겠습니다.(위 첨부된 url에 자세히 기재되어있음) Pytorch 1.11.0 & Torchvision 0.12.0 설치 설치를 진행하기 앞서 필요한 필수 패키지를 설치합니다. sudo apt update sudo apt install -y libopenmpi-dev libomp-dev libopenblas-dev libblas-dev libeigen3-dev libcublas-dev Copy python 패키지를 설치합니다. pip install -U pip wheel gdown Copy pre-built된 pytorch와 torchvision을 다운받아 설치합니다. gdown https://drive.google.com/uc?id=1hs9HM0XJ2LPFghcn7ZMOs5qu5HexPXwM gdown https://drive.google.com/uc?id=1m0d8ruUY8RvCP9eVjZw4Nc8LAwM8yuGV python -m pip install torch-*.whl torchvision-*.whl Copy yolo를 사용하기 위해서 ultralytics 패키지를 설치합니다. pip install ultralytics Copy 이후 yolo를 실행하면 GPU를 이용해서 정상적으로 동작함을 확인할 수 있습니다. jtop 명령어를 통해서 GPU 사용량 확인 가능합니다. 참고 문헌 [1] ...