PyTorch with CuDA support in Jetson Xavier NX Dev kit

I am attempting to install PyTorch with CUDA support. However, even after completing the installation, I am still receiving the following output.

import torch
print (torch.version)
2.0.1
print (torch.cuda.is_available())
false

Result (PyTorch with CUDA support false)
what is the better way to install PyTorch with CUDA support
in Jetson Xavier NX Dev kit

Jetson info

Hi,

Installing PyTorch from the official website may result in failure.
Please refer to this forum to install PyTorch that matches the corresponding JetPack version.

Thanks

Thanks for your help!

1 Like

This topic was automatically closed 14 days after the last reply. New replies are no longer allowed.