Pytorch and CUDA on Jetson Xavier NX

Autonomous Machines Jetson & Embedded Systems Jetson Xavier NX
1

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

3

Hi,

How do you install the PyTorch package.
Could you try our package in the below topic or NGC l4t-pytorch container?

Thanks.

4

Hi Thanks for your response,
I just tried your link. Is there any way doing this without using a container?

5

When I try to install the local version I get the following error ERROR: torch-1.7.0-cp36-cp36m-linux_aarch64.whl is not a supported wheel on this platform. This was using pip3 install torch-1.7.0-cp36-cp36m-linux_aarch64.whlThanks, Charlie

6

Hmm did you install the PyTorch 1.8 wheel from that post that Aasta linked to also? Is this the URL you used for the PyTorch 1.7 wheel?

https://nvidia.box.com/shared/static/cs3xn3td6sfgtene6jdvsxlr366m2dhq.whl

What version of JetPack-L4T are you running? You can check your L4T version with cat /etc/nv_tegra_release

Is there any different if you run deviceQuery sample first, before trying to run PyTorch command?

cd /usr/local/cuda/samples/1_Utilities/deviceQuery
sudo make
./deviceQuery

Does deviceQuery report your GPU? If so, does PyTorch detect the GPU after running deviceQuery?

8