Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Tensorflow GPU support not working #85

Open
Nicholas-B1 opened this issue May 3, 2024 · 7 comments
Open

Tensorflow GPU support not working #85

Nicholas-B1 opened this issue May 3, 2024 · 7 comments

Comments

@Nicholas-B1
Copy link

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

@Qengineering
Copy link
Owner

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
Copy link
Author

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

@Qengineering
Copy link
Owner

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
Copy link
Author

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

@Qengineering
Copy link
Owner

Qengineering commented May 4, 2024

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
Copy link
Author

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
Copy link
Owner

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

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants