Torch.cuda.is_available() returns false

Robotics & Edge Computing Jetson & Embedded Systems Jetson AGX Orin
1

My Jetson Kit environment is below.

  • torch 2.2.2 , torchaudio 2.2.2 , torchvision 0.17.2 , pycuda 2024.1 , python 3.8
  • Jetpack 5.1 , CUDA 11.4 version.

However , when i try to check cuda is availability , like as below , then “false” is returned.

import torch
torch.cuda.is_available()
False
Please check about this issue , if i have to update another Pytorch and Cuda version , and really hope to get the exact solution.
Thanks..

4

Hi,

Please provide commands how do you install pytorch.

Thanks

5

i wrote like as below..

  • pip install torch==2.2.2
    Thanks..
6

Hi,

pip install torch not support in jetson, since package is for desktop gpu not embedded.
Please refer to this topic to install pytorch wheel.

Thanks

7

How about torchvision and torchaudio image ? Are they to use as a same way like as a “install wheel” or available for “pip install torchvision” command.

Thanks for your support…

I did as you commented… however same false i found as well…
Please refers to below..

root@wckim-desktop:/home/wckim/ks/testImage# pip list | grep torch
torch 2.1.0
torchaudio 2.1.0
torchvision 0.16.0
root@wckim-desktop:/home/wckim/ks/testImage#
root@wckim-desktop:/home/wckim/ks/testImage#
root@wckim-desktop:/home/wckim/ks/testImage# python3
Python 3.8.10 (default, Mar 18 2025, 20:04:55)
[GCC 9.4.0] on linux
Type “help”, “copyright”, “credits” or “license” for more information.

import torch
torch.cuda.is_available()
False

Would you please let me know the exact torch / torchvison / torchaudio version that i have to download to match with my configuration environment that i mentioned before…

Thanks..

8

Hi,

Please try below commands:

wget https://pypi.jetson-ai-lab.dev/jp5/cu114/+f/4c1/d7a5d0ba92527/torch-2.2.0-cp38-cp38-linux_aarch64.whl#sha256=4c1d7a5d0ba92527c163ce9da74a2bdccce47541ef09a14d186e413a47337385
wget https://pypi.jetson-ai-lab.dev/jp5/cu114/+f/edd/edbc352a2b140/torchaudio-2.2.2+cefdb36-cp38-cp38-linux_aarch64.whl#sha256=eddedbc352a2b1406f0e4623edb938aebcda50257b523f3d028048af6abae57e
wget https://pypi.jetson-ai-lab.dev/jp5/cu114/+f/12c/2173bcd5255bd/torchvision-0.17.2+c1d70fe-cp38-cp38-linux_aarch64.whl#sha256=12c2173bcd5255bddad13047c573de24e0ce2ea47374c48ee8fb88466e021d2a

pip install torch-2.2.0-cp38-cp38-linux_aarch64.whl
pip install torchaudio-2.2.2+cefdb36-cp38-cp38-linux_aarch64.whl
pip install torchvision-0.17.2+c1d70fe-cp38-cp38-linux_aarch64.whl

Thanks

10

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