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

Still having issues unfortunately #5

Closed AIisCool closed 1 year ago

AIisCool commented 2 years ago

Not sure what to do. I installed jaxlib for Windows via: https://github.com/cloudhan/jax-windows-builder

But I'm having this problem:

$ python maxim/run_eval.py --task Denoising --ckpt_path C:/maxim/Denoising_SIDD.npz \
>   --input_dir C:/maxim/images/Denoising/ --output_dir C:/maxim/images/Results --has_target=False
C:\Python39\lib\site-packages\numpy\_distributor_init.py:30: UserWarning: loaded more than 1 DLL from .libs:
C:\Python39\lib\site-packages\numpy\.libs\libopenblas.EL2C6PLE4ZYW3ECEVIV3OXXGRN2NRFM2.gfortran-win_amd64.dll
C:\Python39\lib\site-packages\numpy\.libs\libopenblas.QVLO2T66WEPI7JZ63PS3HMOHFEY472BC.gfortran-win_amd64.dll
  warnings.warn("loaded more than 1 DLL from .libs:"
I0406 23:21:24.238674 16916 xla_bridge.py:141] Remote TPU is not linked into jax; skipping remote TPU.
I0406 23:21:24.238674 16916 xla_bridge.py:231] Unable to initialize backend 'tpu_driver': Could not initialize backend 'tpu_driver'
I0406 23:21:24.301170 16916 xla_bridge.py:231] Unable to initialize backend 'tpu': INVALID_ARGUMENT: TpuPlatform is not available.
2022-04-06 23:21:24.928930: W external/org_tensorflow/tensorflow/stream_executor/gpu/asm_compiler.cc:111] *** WARNING *** You are using ptxas 10.2.89, which is older than 11.1. ptxas before 11.1 is known to miscompile XLA code, leading to incorrect results or invalid-address errors.

You may not need to update to CUDA 11.1; cherry-picking the ptxas binary is often sufficient.
2022-04-06 23:21:24.981518: F external/org_tensorflow/tensorflow/compiler/xla/service/gpu/nvptx_compiler.cc:435] ptxas returned an error during compilation of ptx to sass: 'INTERNAL: ptxas exited with non-zero error code -1, output: '  If the error message indicates that a file could not be written, please verify that sufficient filesystem space is provided.
Fatal Python error: Aborted

Thread 0x00004214 (most recent call first):
  File "C:\Python39\lib\site-packages\jax\interpreters\xla.py", line 389 in backend_compile
  File "C:\Python39\lib\site-packages\jax\interpreters\xla.py", line 83 in compile_or_get_cached
  File "C:\Python39\lib\site-packages\jax\interpreters\xla.py", line 822 in from_xla_computation
  File "C:\Python39\lib\site-packages\jax\interpreters\xla.py", line 797 in compile
  File "C:\Python39\lib\site-packages\jax\interpreters\xla.py", line 696 in _xla_callable
  File "C:\Python39\lib\site-packages\jax\linear_util.py", line 263 in memoized_fun
  File "C:\Python39\lib\site-packages\jax\interpreters\xla.py", line 624 in _xla_call_impl
  File "C:\Python39\lib\site-packages\jax\core.py", line 631 in process_call
  File "C:\Python39\lib\site-packages\jax\core.py", line 1642 in process
  File "C:\Python39\lib\site-packages\jax\core.py", line 1630 in call_bind
  File "C:\Python39\lib\site-packages\jax\core.py", line 1639 in bind
  File "C:\Python39\lib\site-packages\jax\_src\api.py", line 415 in cache_miss
  File "C:\Python39\lib\site-packages\jax\_src\traceback_util.py", line 162 in reraise_with_filtered_traceback
  File "C:\Python39\lib\site-packages\jax\_src\image\scale.py", line 308 in resize
  File "C:\Python39\lib\site-packages\maxim\models\maxim.py", line 676 in __call__
  File "C:\Python39\lib\site-packages\flax\linen\module.py", line 271 in wrapped_module_method
  File "C:\Python39\lib\site-packages\flax\linen\module.py", line 1135 in scope_fn
  File "C:\Python39\lib\site-packages\flax\core\scope.py", line 675 in wrapper
  File "C:\Python39\lib\site-packages\flax\linen\module.py", line 911 in apply
  File "C:\maxim\maxim\run_eval.py", line 372 in _process_file
  File "C:\maxim\maxim\run_eval.py", line 409 in main
  File "C:\Python39\lib\site-packages\absl\app.py", line 258 in _run_main
  File "C:\Python39\lib\site-packages\absl\app.py", line 312 in run
  File "C:\maxim\maxim\run_eval.py", line 419 in <module>
vztu commented 2 years ago

Sorry I cannot parse these error logs. Looks like a building/compiling error with Jax on Windows but I don't have any experience with Windows debug. It's better to find a Linux system to run the code or use Virtual Machines.