From 7bd6cc13e0641ce774a364c6dd4b298b0c3ea69a Mon Sep 17 00:00:00 2001 From: Peter Hawkins Date: Mon, 12 Dec 2022 11:03:48 -0500 Subject: [PATCH] Note CUDA 11.8 is required for Hopper and Ada Lovelace series GPUs. Fixes https://github.com/google/jax/issues/13570 --- README.md | 3 +++ 1 file changed, 3 insertions(+) diff --git a/README.md b/README.md index 77d1ca7b37db..84adb8ca5ca3 100644 --- a/README.md +++ b/README.md @@ -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.