Skip to content

Commit

Permalink
Merge pull request #13614 from hawkinsp:cuda
Browse files Browse the repository at this point in the history
PiperOrigin-RevId: 494766254
  • Loading branch information
jax authors committed Dec 12, 2022
2 parents b868cf7 + 7bd6cc1 commit 5e8c0ec
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 5e8c0ec

Please sign in to comment.