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


Could you share the package version you installed with us?

>>> tf.__version__



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')]



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


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.


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.

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