Hi,
I’m facing an issue on an Nvidia Jetson Orin Nano where the GPU is not being detected.
nvcc --version command shows that CUDA 12.2 version is installed.
Nvidia jetpack 6.0 is isntalled
Package: nvidia-jetpack
Version: 6.0-b52
Architecture: arm64
Maintainer: NVIDIA Corporation
Installed-Size: 194
Depends: nvidia-jetpack-runtime (= 6.0-b52), nvidia-jetpack-dev (= 6.0-b52)
Homepage: Jetson - Embedded AI Computing Platform | NVIDIA Developer
I have installed pytorch packages on this link which is compatible with Cuda 12.2
https://developer.download.nvidia.com/compute/redist/jp/v60/pytorch/ torch-2.4.0a0+07cecf4168.nv24.05.14710581-cp310-cp310-linux_aarch64.whl
Sudo jtop shows the gpu
Other libraries version are here
OpenCv version 4.9.0[installed]
Pytorch Version 2.3.1[installed]
libcudnn8/stable,now 8.9.4.25-1+cuda12.2 arm64 [installed]
libcudnn8-dev/stable,now 8.9.4.25-1+cuda12.2 arm64 [installed]
libcudnn8-samples/stable,now 8.9.4.25-1+cuda12.2 arm64 [installed]
But PyTorch says that cuda is not available:
Hello,
Welcome to the NVIDIA Developer forums! Your topic will be best served in the Jetson category.
Discussions relating to the Jetson DevKits and other Embedded computing devices
I will move this post over for visibility.
Cheers,
Tom
1 Like
Hi,
It doesn’t mean GPU is not available.
Usually, PyTorch means the package isn’t built with CUDA support.
To solve this, please re-install PyTorch with our package that has been built with CUDA:
Below are pre-built PyTorch pip wheel installers for Jetson Nano, TX1/TX2, Xavier, and Orin with JetPack 4.2 and newer.
Download one of the PyTorch binaries from below for your version of JetPack, and see the installation instructions to run on your Jetson. These pip wheels are built for ARM aarch64 architecture, so run these commands on your Jetson (not on a host PC). You can also use the containers from jetson-containers .
PyTorch pip wheels
JetPack 6
PyTorch v2.3.0 JetPack 6.0 (L4T R36.2…
Thanks.
system
Closed
July 31, 2024, 3:56am
7
This topic was automatically closed 14 days after the last reply. New replies are no longer allowed.