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.
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.