Nothing Special   »   [go: up one dir, main page]
Skip to content
This repository has been archived by the owner on Aug 12, 2024. It is now read-only.

No GPU/TPU found error #42

Open
sunhoro opened this issue Jul 17, 2023 · 2 comments
Open

No GPU/TPU found error #42

sunhoro opened this issue Jul 17, 2023 · 2 comments

Comments

@sunhoro
Copy link
sunhoro commented Jul 17, 2023

Any help would be appreciated.
I am running on two PC with Windows 11 and Windows 10, each with Nvidia GPU, and facing same issue on both.

All installation was done correctly but keep getting "No GPU/TPU found, falling back to CPU" error which still runs fine but super slow.
I think cuda and jax might not be compatible at this point but not sure.

image

I made sure everything is up to date including pip, cuda, and jax but still the same issue.

  • cuda = 12.2
  • jax = 0.4.13
  • pip = 23.1.2
@Sk-bishnoi
Copy link

same error getting, if you got @sunhoro solution please let me know
No GPU/TPU found, falling back to CPU. (Set TF_CPP_MIN_LOG_LEVEL=0 and rerun for more info.)
output/1.png
final output/1.png

@njustlxk
Copy link

You need to install the GPU version of JAX, which is often not compatible with Windows

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
3 participants
@sunhoro @Sk-bishnoi @njustlxk