I’m trying to install the TRTorch on my brand new Xavier AGX, flashed with JetPack 4.4.
The problem is that TRTorch supports pytorch 1.5.1 and we don’t have the pytorch 1.5.1 available for Jetson. I checked the l4t-pytorch docker containers and we just have 1.5.0 and 1.6.0. Both fail, as cuda version is different (1.5.0) and header files changed (for 1.6.0).
Do you know how to install pytorch 1.5.1 in xavier agx ?
Is some change in progress to make it work on the 1.6.0?