Closed giotherobot closed 8 months ago
To have a more self-contained example, this fails:
traversaro@IITBMP014LW012:~$ conda create -n kerastest "keras==3.0.5" "jax==0.4.23"
Retrieving notices: ...working... done
Channels:
- conda-forge
Platform: linux-64
Collecting package metadata (repodata.json): done
Solving environment: failed
LibMambaUnsatisfiableError: Encountered problems while solving:
- package keras-3.0.5-pyhd8ed1ab_0 has constraint jax 0.4 conflicting with jax-0.4.23-pyhd8ed1ab_0
Could not solve for environment specs
The following packages are incompatible
├─ jax 0.4.23 is requested and can be installed;
└─ keras 3.0.5 is not installable because it requires
└─ jax 0.4 , which conflicts with any installable versions previously reported.
Solution to issue cannot be found in the documentation.
Issue
Trying to create a gpu environment with CUDA 12, keras 3 and jax backend fails.
Tested with:
Environment.yaml
``` name: testenv channels: - conda-forge dependencies: - jax>0.4.23 - jaxlib - cuda-version=12.* - cuda-cupti - cuda-nvcc - keras>=3.0.0 ```
Solver output:
jax >=0.4.23 is required to obtain jaxlib compiled for cuda 12.
Installed packages
Environment info