Jetson Xavier Nx - Cuda toolkit

Autonomous Machines Jetson & Embedded Systems Jetson Xavier NX
, ,
1

Hi guys, I have Jetson Xavier Nx. I’m working on a project that using Yolo. My program need Torch & Torchvision . I already installed torch>=1.7.0, torchvision>=0.8.1
but when I run torch.cuda.is_Available() is returning False. Older Cuda versions working fine but I need 1.7.0 or newer versions.
How could you help me?

2

Hi @hisisman3, if you installed PyTorch/torchvision from pip (or if installing a yolo package from pip pulled those in as dependencies), then the torch wheels from PyPi aren’t built with CUDA enabled. These PyTorch wheels were built with CUDA enabled:

There are also instructions in that post for building torchvision with CUDA. First you should pip3 uninstall torch torchvision to remove the versions that don’t have GPU acceleration.

4

(moving this post to the Xavier NX forum)

6

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