optax error when installing on Windows

[SoftologyPro]

Hello,

I create a new Python environment and activate it.
Then I try pip install dalle-mini
After a while it errors out with

INFO: pip is looking at multiple versions of optax to determine which version is compatible with other requirements. This could take a while.
Collecting optax (from flax==0.6.3->dalle-mini)
  Using cached optax-0.1.3-py3-none-any.whl (145 kB)
  Using cached optax-0.1.2-py3-none-any.whl (140 kB)
  Using cached optax-0.1.1-py3-none-any.whl (136 kB)
  Using cached optax-0.1.0-py3-none-any.whl (126 kB)
  Using cached optax-0.0.91-py3-none-any.whl (126 kB)
  Using cached optax-0.0.9-py3-none-any.whl (118 kB)
  Using cached optax-0.0.8-py3-none-any.whl (113 kB)
INFO: pip is looking at multiple versions of optax to determine which version is compatible with other requirements. This could take a while.
  Using cached optax-0.0.6-py3-none-any.whl (96 kB)
  Using cached optax-0.0.5-py3-none-any.whl (89 kB)
  Using cached optax-0.0.3-py3-none-any.whl (73 kB)
  Using cached optax-0.0.2-py3-none-any.whl (54 kB)
  Using cached optax-0.0.1-py3-none-any.whl (49 kB)
INFO: This is taking longer than usual. You might need to provide the dependency resolver with stricter constraints to reduce runtime. See https://pip.pypa.io/warnings/backtracking for guidance. If you want to abort this run, press Ctrl + C.
ERROR: Cannot install flax because these package versions have conflicting dependencies.

The conflict is caused by:
    optax 0.1.4 depends on jaxlib>=0.1.37
    optax 0.1.3 depends on jaxlib>=0.1.37
    optax 0.1.2 depends on jaxlib>=0.1.37
    optax 0.1.1 depends on jaxlib>=0.1.37
    optax 0.1.0 depends on jaxlib>=0.1.37
    optax 0.0.91 depends on jaxlib>=0.1.37
    optax 0.0.9 depends on jaxlib>=0.1.37
    optax 0.0.8 depends on jaxlib>=0.1.37
    optax 0.0.6 depends on jaxlib>=0.1.37
    optax 0.0.5 depends on jaxlib>=0.1.37
    optax 0.0.3 depends on jaxlib>=0.1.37
    optax 0.0.2 depends on jaxlib>=0.1.37
    optax 0.0.1 depends on jaxlib>=0.1.37

To fix this you could try to:
1. loosen the range of package versions you've specified
2. remove package versions to allow pip attempt to solve the dependency conflict

ERROR: ResolutionImpossible: for help visit https://pip.pypa.io/en/latest/topics/dependency-resolution/#dealing-with-dependency-conflicts

Any fix for this? Thanks.

[codespearhead]

What's the output of python -V && pip -V?

What about the output of the following?

python <<< "from platform import uname;print(uname())"

[SoftologyPro]

Python 3.10.8
pip 23.1.1 from D:\Python\lib\site-packages\pip (python 3.10)
uname_result(system='Windows', node='PCNAMEHERE', release='10', version='10.0.22621', machine='AMD64')

[SoftologyPro]

Any fix for this one?

[borisdayma]

Try again maybe with newest release. I've not seen this error before.