No GPU/TPU found error - Githubissues

google-research / maxim

[CVPR 2022 Oral] Official repository for "MAXIM: Multi-Axis MLP for Image Processing". SOTA for denoising, deblurring, deraining, dehazing, and enhancement.

https://arxiv.org/abs/2201.02973

Apache License 2.0

974 stars 105 forks source link

No GPU/TPU found error #42

Open sunhoro opened 11 months ago

sunhoro commented 11 months ago

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.

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 commented 11 months ago

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 commented 1 month ago

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