Giters

google-research / maxim

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

Home Page:https://arxiv.org/abs/2201.02973

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

No GPU found

hs0817 opened this issue · comments

commented

Hello, I installed the environment and projects according to requirements. txt, but there was no GPU in the test process. My CUDA version was 11.2, and the GPU was RTX3090. Can you help me explain this situation?
1659858182299

commented

You should use the jaxlib's cuda version.

commented

Thank you for your quick response.