Skip to content

Commit

Permalink
Note CUDA 11.8 is required for Hopper and Ada Lovelace series GPUs.
Browse files Browse the repository at this point in the history
Fixes #13570
  • Loading branch information
hawkinsp committed Dec 12, 2022
1 parent 13c34f9 commit 7bd6cc1
Showing 1 changed file with 3 additions and 0 deletions.
3 changes: 3 additions & 0 deletions README.md
Expand Up @@ -432,6 +432,9 @@ operating system, CUDA, and CuDNN are possible, but require [building from
source](https://jax.readthedocs.io/en/latest/developer.html#building-from-source).

* CUDA 11.4 or newer is *required*.
* Your CUDA installation must be new enough to support your GPU. If you have
an Ada Lovelace (e.g., RTX 4080) or Hopper (e.g., H100) GPU,
you must use CUDA 11.8 or newer.
* The supported cuDNN versions for the prebuilt wheels are:
* cuDNN 8.6 or newer. We recommend using the cuDNN 8.6 wheel if your cuDNN
installation is new enough, since it supports additional functionality.
Expand Down

0 comments on commit 7bd6cc1

Please sign in to comment.