Jetson 的 PyTorch安装
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 --branch
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的版本对应关系
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
原文链接:https://forums.developer.nvidia.com/t/pytorch-for-jetson-version-1-9-0-now-available/72048