Tensorflow is not able to detect gpu on Orin

jay.parikh1 · March 9, 2023, 3:45pm

I installed Tensorflow using this guide

Using Jetpack version 5.1, Please find attached screenshot

AastaLLL · March 10, 2023, 2:59am

Hi,

Could you share the package version you installed with us?

>>> tf.__version__

Thanks.

AastaLLL · March 10, 2023, 3:31am

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.

jay.parikh1 · March 10, 2023, 3:50am

Hi

It’s 2.11 and nv23.01

jay.parikh1 · March 10, 2023, 3:52am

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

AastaLLL · March 13, 2023, 5:18am

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.

jay.parikh1 · March 13, 2023, 2:12pm

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

AastaLLL · March 14, 2023, 4:21am

Good to know this.
Thanks for the update.