Jetson Orin Nx, Pytorch 2.1.2 on Jetpack 6.0 DP, torch.cuda.is_available() is false

Hi all,
My hardware is Jetson Orin NX, and I have installed JetPack 6.0 DP. I created a virtual environment using Python 3.8.17 and used the command conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia to install PyTorch version 2.1.2.

In the Python interpreter, when using torch.__version__ , it returns ‘2.1.2’. However, when using torch.cuda.is_available() , it returns false. How can I make torch.cuda.is_available() return true?

Thank you


The packages on the apt/pip server are only built with CPU support.
Please use our prebuilt instead to get the CUDA support:


1 Like

Hi AastaLLL,

I use Jetpack 6.0 DP,and Pytorch 2.2.0


it work

Thank you

Hi AastaLLL,

I install Pytorch 2.2.0,

and I need to install which version of torchvision.



Please try TorchVision 0.17.


Hi AasTaLLL,

I download torchvision 0.17 from this website

and use command $ sudo pip3 install torchvision-0.17.0.dev20231211-cp310-cp310-linux_aarch64.whl

but return error message :

ERROR: Could not find a version that satisfies the requirement torch==2.3.0.dev20231211 (from torchvision) (from versions: 1.10.2, 1.11.0, 1.12.0, 1.12.1, 1.13.0, 1.13.1, 2.0.0, 2.0.1, 2.1.0, 2.1.1, 2.1.2)
ERROR: No matching distribution found for torch==2.3.0.dev20231211

and i try torchvision-0.17.0.dev20231010-cp310-cp310-linux_aarch64.whl

return ERROR: No matching distribution found for torch==2.2.0.dev20231010

How can I fix it?


@cho56915 these wheels from website were not built with CUDA enabled, they are for general ARM64 platform without GPU. Instead, please build torchvision from source following the torchvision instructions under the Installation section of this topic:

Or you can use this container which already has it:

That has PyTorch/torchvision/torchaudio/ect already installed, and the compiled wheels are saved under /opt inside the container if you want to copy them out.

Hi dusty_nv

The issue has been resolved.

Thank you for your suggestions.

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