How to install torchvision0.8.0 in Jetson TX2 (Jetpack4.5.1,pytorch1.7.0)

I’ve installed pytorch1.7.0,how to install torchvision?

Hi,

You can find the instruction in the below topic:
https://forums.developer.nvidia.com/t/pytorch-for-jetson-version-1-9-0-now-available/72048

$ sudo apt-get install libjpeg-dev zlib1g-dev libpython3-dev libavcodec-dev libavformat-dev libswscale-dev
$ git clone --branch <version> https://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

Thanks.

1 Like

Hi,I find that pycharm linux verision doesn’t support TX2,can you please tell me a project IDE software or any project development guide so that I can get familiar with TX2 quickly?