<h3 dir="auto">Description</h3>
<p dir="auto">I am unable to install the latest version of JAX 0.4.12 through the command:</p>
<div class="snippet-clipboard-content notranslate position-relative overflow-auto" data-snippet-clipboard-copy-content="pip install --upgrade &quot;jax[cuda12_pip]&quot; -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html"><pre class="notranslate"><code class="notranslate">pip install --upgrade "jax[cuda12_pip]" -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
</code></pre></div>
<p dir="auto">Version 0.4.11 is installed instead.</p>
<h3 dir="auto">What jax/jaxlib version are you using?</h3>
<p dir="auto"><em>No response</em></p>
<h3 dir="auto">Which accelerator(s) are you using?</h3>
<p dir="auto"><em>No response</em></p>
<h3 dir="auto">Additional system info</h3>
<p dir="auto"><em>No response</em></p>
<h3 dir="auto">NVIDIA GPU info</h3>
<p dir="auto">+---------------------------------------------------------------------------------------+<br>
| NVIDIA-SMI 530.30.02              Driver Version: 530.30.02    CUDA Version: 12.1     |<br>
|-----------------------------------------+----------------------+----------------------+<br>
| GPU  Name                  Persistence-M| Bus-Id        Disp.A | Volatile Uncorr. ECC |<br>
| Fan  Temp  Perf            Pwr:Usage/Cap|         Memory-Usage | GPU-Util  Compute M. |<br>
|                                         |                      |               MIG M. |<br>
|=========================================+======================+======================|<br>
|   0  NVIDIA RTX 6000 Ada Gener...    Off| 00000000:41:00.0 Off |                  Off |<br>
| 30%   44C    P8               26W / 300W|      6MiB / 49140MiB |      0%      Default |<br>
|                                         |                      |                  N/A |<br>
+-----------------------------------------+----------------------+----------------------+<br>
|   1  NVIDIA RTX 6000 Ada Gener...    Off| 00000000:61:00.0 Off |                  Off |<br>
| 30%   39C    P8               22W / 300W|    226MiB / 49140MiB |      0%      Default |<br>
|                                         |                      |                  N/A |<br>
+-----------------------------------------+----------------------+----------------------+</p>
<p dir="auto">+---------------------------------------------------------------------------------------+<br>
| Processes:                                                                            |<br>
|  GPU   GI   CI        PID   Type   Process name                            GPU Memory |<br>
|        ID   ID                                                             Usage      |<br>
|=======================================================================================|<br>
|    0   N/A  N/A      1925      G   /usr/lib/xorg/Xorg                            4MiB |<br>
|    1   N/A  N/A      1925      G   /usr/lib/xorg/Xorg                          135MiB |<br>
|    1   N/A  N/A      2061      G   /usr/bin/gnome-shell                         52MiB |<br>
+---------------------------------------------------------------------------------------+</p>
<h3 dir="auto">Description</h3>
<p dir="auto">For CUDA 12.1 and cuDNN 8.8 No wheel is available. The suggested whl to use was CUDA 12 and cuDNN 8.9 but this does not work and I only have cuDNN 8.8.</p>
<p dir="auto">Thanks,</p>
<h3 dir="auto">What jax/jaxlib version are you using?</h3>
<p dir="auto">jaxlib==0.4.12</p>
<h3 dir="auto">Which accelerator(s) are you using?</h3>
<p dir="auto">GPU</p>
<h3 dir="auto">Additional system info</h3>
<p dir="auto">Python 3.8.10, Ubuntu 20.04.6 LTS</p>
<h3 dir="auto">NVIDIA GPU info</h3>
<p dir="auto">Mon Jun 12 14:37:26 2023<br>
+---------------------------------------------------------------------------------------+<br>
| NVIDIA-SMI 530.41.03              Driver Version: 530.41.03    CUDA Version: 12.1     |<br>
|-----------------------------------------+----------------------+----------------------+<br>
| GPU  Name                  Persistence-M| Bus-Id        Disp.A | Volatile Uncorr. ECC |<br>
| Fan  Temp  Perf            Pwr:Usage/Cap|         Memory-Usage | GPU-Util  Compute M. |<br>
|                                         |                      |               MIG M. |<br>
|=========================================+======================+======================|<br>
|   0  NVIDIA RTX A6000                Off| 00000000:65:00.0  On |                  Off |<br>
| 32%   61C    P2               79W / 300W|  48625MiB / 49140MiB |      0%      Default |<br>
|                                         |                      |                  N/A |<br>
+-----------------------------------------+----------------------+----------------------+</p>
<p dir="auto">+---------------------------------------------------------------------------------------+<br>
| Processes:                                                                            |<br>
|  GPU   GI   CI        PID   Type   Process name                            GPU Memory |<br>
|        ID   ID                                                             Usage      |<br>
|=======================================================================================|<br>
|    0   N/A  N/A      1189      G   /usr/lib/xorg/Xorg                           77MiB |<br>
|    0   N/A  N/A      1939      G   /usr/lib/xorg/Xorg                          161MiB |<br>
|    0   N/A  N/A      2074      G   /usr/bin/gnome-shell                         75MiB |<br>
|    0   N/A  N/A      6749      C   python3                                   37240MiB |<br>
|    0   N/A  N/A      7046      C   python3                                   10934MiB |<br>
|    0   N/A  N/A     11541      G   /usr/lib/firefox/firefox                    121MiB |<br>
+---------------------------------------------------------------------------------------+</p>