Is there a stable way to install/build Torch-Tensorrt for Jetson Xavier NX with Jetpack 5.1.2? Kindly help
Hi @karthika1 ,
I have not built Torch and TRT on Jetpack 5.1.2, but the process is the same as here, just update with the respective matching versions for CUDA, TRT and Pythorch. For Jetpack 5.1.2 the CUDA version is 11.4.19, TensorRT is 8.5.2. For Torch it will depend on if you have an specific model need, that requires a specific version or not. Also for reference you can check here, they tried it for Jetpack 5.1.
Thanks @andres.artavia - I also have a container built and Dockerfile for it on JetPack 5: