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
1.02k stars 110 forks source link

questions about run_eval.py #31

Open chaolistat opened 2 years ago

chaolistat commented 2 years ago

Hello, excellent work! But I have the following problems in operation. Can you answer them?thanks

2022-10-21 18:23:03.523703: E tensorflow/stream_executor/cuda/cuda_blas.cc:2981] Unable to register cuBLAS factory: Attempting to register factory for plugin cuBLAS when one has already been registered 2022-10-21 18:23:04.678118: W tensorflow/stream_executor/platform/default/dso_loader.cc:64] Could not load dynamic library 'libnvinfer.so.7'; dlerror: libnvinfer.so.7: cannot open shared object file: No such file or directory; LD_LIBRARY_PATH: :/usr/local/cuda/lib64 2022-10-21 18:23:04.678288: W tensorflow/stream_executor/platform/default/dso_loader.cc:64] Could not load dynamic library 'libnvinfer_plugin.so.7'; dlerror: libnvinfer_plugin.so.7: cannot open shared object file: No such file or directory; LD_LIBRARY_PATH: :/usr/local/cuda/lib64 2022-10-21 18:23:04.678303: W tensorflow/compiler/tf2tensorrt/utils/py_utils.cc:38] TF-TRT Warning: Cannot dlopen some TensorRT libraries. If you would like to use Nvidia GPU with TensorRT, please make sure the missing libraries mentioned above are installed properly. Processing 1 / 65... I1021 18:23:51.411981 140692963194624 xla_bridge.py:357] Unable to initialize backend 'tpu_driver': NOT_FOUND: Unable to find driver in registry given worker: I1021 18:23:51.413208 140692963194624 xla_bridge.py:357] Unable to initialize backend 'cuda': module 'jaxlib.xla_extension' has no attribute 'GpuAllocatorConfig' I1021 18:23:51.413419 140692963194624 xla_bridge.py:357] Unable to initialize backend 'rocm': module 'jaxlib.xla_extension' has no attribute 'GpuAllocatorConfig' I1021 18:23:51.453943 140692963194624 xla_bridge.py:357] Unable to initialize backend 'tpu': INVALID_ARGUMENT: TpuPlatform is not available. I1021 18:23:51.454329 140692963194624 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. W1021 18:23:51.454519 140692963194624 xla_bridge.py:364] No GPU/TPU found, falling back to CPU. (Set TF_CPP_MIN_LOG_LEVEL=0 and rerun for more info.) Killed

YoungofNUAA commented 2 years ago

i have the same problem have you solved it yet