Error with Jax installation missing 'requirements-gpu.txt'

[Chi-JuiChen]

While attempting to install Jax, following the provided instructions in the installation guide, I encountered an error when executing the command pip install -r requirements-gpu.txt -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html. The error message received was: "ERROR: Could not open requirements file: [Errno 2] No such file or directory: 'requirements-gpu.txt'".

It seems like Google may have changed the data structure, resulting in the absence of the requirements-gpu.txt file. This prevents users from installing the required dependencies for GPU usage as intended.

[kasProg]

use pip install --upgrade "jax[cuda12_pip]" -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html

[Chi-JuiChen]

Thanks. By the way, the newest version, JAX 0.4.26, will not work. Limiting it to 0.4.23 works.