How to install torchvision with torch1.14.0 with CUDA 11.4

Hi @dgotfroi ,
Usually if the packages are not readilly available is best just to build from source. And the bonus is that sometimes it gives an extra bump in performance.
Here is the source repo and according to them you just need to have Torch installed and do:

python setup.py install

Before doing so I suggest to enable jetson-clocks to speed things up and enable all cores.
There are some ways to do that, here is one using cmd, and another is using jtop.

Regards,
Andres
Embedded SW Engineer at RidgeRun
Contact us: support@ridgerun.com
Developers wiki: https://developer.ridgerun.com/
Website: www.ridgerun.com

2 Likes