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?


You can find the instruction in the below topic:

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


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?

This topic was automatically closed 60 days after the last reply. New replies are no longer allowed.