Hi,
torchvision for Jetson needs to be built from the source.
For example:
$ sudo apt-get install libjpeg-dev zlib1g-dev python3-pip
$ git clone -b v0.7.0 https://github.com/pytorch/vision torchvision
$ export BUILD_VERSION=0.7.0
$ sudo python3 setup.py install
For more information, please check this comment:
Thanks.
1 Like