Hi,
I am trying to run pytorch on my Jetson Xavier NX. I have CUDA 10.2 and have installed pytorch 1.8 for CUDA 10.2 but pytorch does not recognize that I have CUDA available with the following command:
print('CUDA available: ' + str(torch.cuda.is_available()))
It would be great if anyone could help me work out why its not working.
C