常见问题

Jetson 的 PyTorch安装

2021-09-29 17:48:59 admin 28

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的版本对应关系

  • 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

首页
产品
案例
联系