How to install torchvision 0.9.0?

Robotics & Edge Computing Jetson & Embedded Systems Jetson AGX Xavier
3

Hi,

Since Jetson is an ARM system, there is no torchvision prebuilt available from the pip installer.
You can build it from source as below:

$ sudo apt-get install libjpeg-dev zlib1g-dev libpython3-dev libavcodec-dev libavformat-dev libswscale-dev
$ git clone --branch v0.9.0 https://github.com/pytorch/vision torchvision
$ cd torchvision
$ export BUILD_VERSION=0.9.0
$ python3 setup.py install --user

Thanks.

1 Like