Tensorflow GPU not detected

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

I am trying to install tensorflow with GPU support on the Orin Nano. I have followed the instructions here: Installing TensorFlow for Jetson Platform - NVIDIA Docs

However, tensorflow is not able to detect the Jetson GPU.

print("Num GPUs Available: ", len(tf.config.experimental.list_physical_devices(‘GPU’)))
Num GPUs Available: 0

I have made sure that there is just one version of TF installed. I have also tried versions 2.17, 2.16, 2.15 for JetPack 6.0. Any idea how to fix this?

1 Like
2

Update: I reinstalled 2.15 from scratch again. This time, the GPU is detected, but I get these warnings.

import tensorflow as tf
2024-07-04 08:30:28.028028: E external/local_xla/xla/stream_executor/cuda/cuda_dnn.cc:9373] Unable to register cuDNN factory: Attempting to register factory for plugin cuDNN when one has already been registered
2024-07-04 08:30:28.028148: E external/local_xla/xla/stream_executor/cuda/cuda_fft.cc:607] Unable to register cuFFT factory: Attempting to register factory for plugin cuFFT when one has already been registered
2024-07-04 08:30:28.030501: E external/local_xla/xla/stream_executor/cuda/cuda_blas.cc:1534] Unable to register cuBLAS factory: Attempting to register factory for plugin cuBLAS when one has already been registered

does this mean tensorflow will not be able to use cuDNN and cuBLAS?

4

Hi,

This is a known issue from the TensorFlow library itself.
Please find below the link for more info:

Thanks.

6

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