Jetson 的 PyTorch安装

2022-04-27 admin

Python 3.6

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

sudo apt-get install python3-pip libopenblas-base libopenmpi-dev pip3 install Cythonpip3 install numpy torch-1.8.0-cp36-cp36m-linux_aarch64.whl

Python 2.7

wget https://nvidia.box.com/shared/static/1v2cc4ro6zvsbu0p8h6qcuaqco1qcsif.whl -O torch-1.4.0-cp27-cp27mu-linux_aarch64.whl

sudo apt-get install libopenblas-base libopenmpi-dev pip install future torch-1.4.0-cp27-cp27mu-linux_aarch64.whl

torchvision

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

$ git clone --branchhttps://github.com/pytorch/vision torchvision # see below for version of torchvision to download

$ cd torchvision$ export BUILD_VERSION=0.x.0# where 0.x.0 is the torchvision version

$ python3 setup.py install --user$ cd ../# attempting to load torchvision from build dir will result in import error

$ pip install 'pillow<7'# always needed for Python 2.7, not needed torchvision v0.5.0+ with Python 3.6

Pytorch与torchvision的版本对应关系

原文链接:https://forums.developer.nvidia.com/t/pytorch-for-jetson-version-1-9-0-now-available/72048