Tensorflow is not able to detect gpu on Orin

I installed Tensorflow using this guide

Installing TensorFlow for Jetson Platform

Using Jetpack version 5.1, Please find attached screenshot

Hi,

Could you share the package version you installed with us?

>>> tf.__version__

Thanks.

Hi,

We have verified the 2.11.0+nv23.01 package on Orin+JetPack 5.1
TensorFlow can detect GPU correctly.

>>> tf.config.list_physical_devices('GPU')
[PhysicalDevice(name='/physical_device:GPU:0', device_type='GPU')]

Thanks.

Hi

It’s 2.11 and nv23.01

Do you have debugging tips to find out what’s wrong? I have no idea how to proceed with this on Orin

Hi,

Do you get the output of tf.__version__?
Could you share it with us first?

A common cause is that there are multiple TensorFlow installed, and some are not \built with CUDA.

Thanks.

I removed all pip packages and installed it again. it is working now. Thank you for your help.

Good to know this.
Thanks for the update.