Tensorflow with gpu on orin nano

I recently bought a jetson orin nano and am trying to run a tensorflow model on it. after going throught the installation process from the tensorflow website (tensorflow[and-cuda]), the command “tf.config.list_physical_devices(‘GPU’)” is an empty list, any fixes?

Hi,

You can build it from the source with the repo below:

Thanks.

Hello
Thank you for the fast reply.
Can it be done outside of a container or is this the only way?

Hi,

You can also build the package natively without using a container.
The steps can be found in the script below:

Thanks.

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