Still having issues unfortunately
AIisCool opened this issue · comments
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>
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.