팁&테크

[Jetson] Jetson보드에서 pytorch 설치 방법

承寶(참참) 2023. 8. 2. 00:33
반응형

https://forums.developer.nvidia.com/t/pytorch-for-jetson/72048

 

PyTorch for Jetson

Below are pre-built PyTorch pip wheel installers for Jetson Nano, TX1/TX2, Xavier, and Orin with JetPack 4.2 and newer. Download one of the PyTorch binaries from below for your version of JetPack, and see the installation instructions to run on your Jetson

forums.developer.nvidia.com

 

해당 페이지에서 본인의 jetpack 버전에 맞는 pytorch 배포본 다운로드 할것

 

>> wget https://nvidia.box.com/shared/static/p57jwntv436lfrd78inwl7iml6p13fzh.whl -O torch-1.10.0-cp36-cp36m-linux_aarch64.whl

>> sudo apt-get install python3-pip libopenblas-base libopenmpi-dev libomp-dev

>> pip3 install Cython

>> pip3 install numpy torch-1.10.0-cp36-cp36m-linux_aarch64.whl

 

설치 후 적합한 torchvision 버전을 위 페이지에서 다시 찾을 것

 

  • PyTorch v1.0 - torchvision v0.2.2
  • PyTorch v1.1 - torchvision v0.3.0
  • PyTorch v1.2 - torchvision v0.4.0
  • PyTorch v1.3 - torchvision v0.4.2
  • PyTorch v1.4 - torchvision v0.5.0
  • PyTorch v1.5 - torchvision v0.6.0
  • PyTorch v1.6 - torchvision v0.7.0
  • PyTorch v1.7 - torchvision v0.8.1
  • PyTorch v1.8 - torchvision v0.9.0
  • PyTorch v1.9 - torchvision v0.10.0
  • PyTorch v1.10 - torchvision v0.11.1
  • PyTorch v1.11 - torchvision v0.12.0
  • PyTorch v1.12 - torchvision v0.13.0
  • PyTorch v1.13 - torchvision v0.13.0
  • PyTorch v1.14 - torchvision v0.14.1
  • PyTorch v2.0.0 - torchvision v0.15.1

torchvision 설치 (python 3.6)

 

$ sudo apt-get install libjpeg-dev zlib1g-dev libpython3-dev libopenblas-dev libavcodec-dev libavformat-dev libswscale-dev

$ git clone --branch v.0.xx.x  https://github.com/pytorch/vision torchvision 

$ cd torchvision

$ export BUILD_VERSION=0.xx.x

$ python3 setup.py install --user

$ cd ..

 

 

 

 

 

 

반응형