search

Qengineering / Jetson-Nano-Ubuntu-20-image

Jetson Nano with Ubuntu 20.04 image
https://qengineering.eu/install-ubuntu-20.04-on-jetson-nano.html
BSD 3-Clause "New" or "Revised" License
646 stars 70 forks source link

Tensorflow GPU support not working #85

Open Nicholas-B1 opened 2 months ago

Nicholas-B1 commented 2 months ago

Hello, I installed this OS for my Jetson Nano but I cannot get Tensorflow to recognize my GPUs being installed on the board.

Qengineering commented 2 months ago

Please note that only TensorFlow 2.4.1 or a lower version will work with the Jetson Nano. Above requires CUDA 11, which is not available for the Jetson Nano.

Nicholas-B1 commented 2 months ago

I have tensorflow 2.13. Is there anything else I need for configuring GPU support with this flashed OS?

Qengineering commented 2 months ago

You cannot run TF 2.13 on a Jetson Nano other than in the CPU mode. If you want GPU, you need to downgrade TF to version 2.4 or lower.

Nicholas-B1 commented 2 months ago

I cannot find version of tensorflow that old with the python version that comes installed on this flashed OS.

Qengineering commented 2 months ago

When you first flashed this image, it came with TF 2.4.1 + GPU support installed. Why not start over again, with a new flash?

Nicholas-B1 commented 2 months ago

Okay thank you I can try that. So the flashed image should be able to recognize the jetson nanos GPU right away or is there any configuration needed?

Qengineering commented 2 months ago

It recognize the CUDA cores out of the box. There is no need for additional configurations.