AssertionError: Torch not compiled with CUDA enabled

Hi @sarodeanil35, did you install one of the PyTorch wheels from this topic that was built with CUDA enabled?

My guess is not, since your PyTorch version is 2.1.2, and we haven’t released a PyTorch wheel for JetPack 6 with that specific version.