search

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
977 stars 105 forks source link

No GPU Using #32

Closed DH-HAN closed 1 year ago

DH-HAN commented 1 year ago

I1025 08:07:07.369448 139768635004736 xla_bridge.py:357] Unable to initialize backend 'tpu_driver': NOT_FOUND: Unable to find driver in registry given worker: I1025 08:07:07.369699 139768635004736 xla_bridge.py:357] Unable to initialize backend 'cuda': module 'jaxlib.xla_extension' has no attribute 'GpuAllocatorConfig' I1025 08:07:07.369804 139768635004736 xla_bridge.py:357] Unable to initialize backend 'rocm': module 'jaxlib.xla_extension' has no attribute 'GpuAllocatorConfig' I1025 08:07:07.370839 139768635004736 xla_bridge.py:357] Unable to initialize backend 'tpu': INVALID_ARGUMENT: TpuPlatform is not available. I1025 08:07:07.370978 139768635004736 xla_bridge.py:357] Unable to initialize backend 'plugin': xla_extension has no attributes named get_plugin_device_client. Compile TensorFlow with //tensorflow/compiler/xla/python:enable_plugin_device set to true (defaults to false) to enable this. W1025 08:07:07.371087 139768635004736 xla_bridge.py:364] No GPU/TPU found, falling back to CPU. (Set TF_CPP_MIN_LOG_LEVEL=0 and rerun for more info.)

I set up my environment following your guide, inclue jax, jax lib but This kind of error message come out, and i automatically inference image by using only gpu

are there someone who have same issue?

DH-HAN commented 1 year ago

I solve this issue by installing Right version of JAX thank you for your nice sharing!