Skip to content

Commit 45fd0e2

Browse files
author
Alexander Ororbia
committed
added compat note for pascal gpus and cuda12+jax
1 parent 9081fd6 commit 45fd0e2

1 file changed

Lines changed: 19 additions & 0 deletions

File tree

README.md

Lines changed: 19 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -45,6 +45,25 @@ Matplotlib (>=3.8.0) and imageio (>=2.31.5) and both plotting and density estima
4545
tools (routines within ``ngclearn.utils.density``) will require Scikit-learn (>=0.24.2).
4646
Many of the tutorials will require Matplotlib (>=3.8.0), imageio (>=2.31.5), and Scikit-learn (>=0.24.2).
4747

48+
<i>Note</i>: If you are working with Cuda 12 and want to use jax/jaxlib versions > 0.4.28, you might need to
49+
check that you are working with the right version of Cudnn (e.g., `nvidia-cudnn-cu12==9.10.2.21`) to ensure
50+
that all of ngc-learn's internal supported tools, like in-built convolution/deconvolution, compile
51+
correctly onto the GPU (if using an architecture based on Pascal GPUs, i.e., Compute Capability 6.1,
52+
combined with NVIDIA Driver 580+).
53+
54+
**Important Note for Legacy GPU Users (Pascal Architecture)**
55+
> If you are running JAX (`> 0.4.28`) on **CUDA 12** using an older **Pascal-generation GPU** (Compute Capability 6.1,
56+
> e.g., GTX 1080/1080Ti, Titan X) combined with **NVIDIA Driver 580+**, you might encounter compilation crashes during
57+
> convolution/deconvolution operations (such as `unknown cudnn status: 5003`).
58+
>
59+
> Newer versions of `nvidia-cudnn-cu12` have dropped critical hardware support for these legacy architectures. To
60+
> fix this and ensure `ngclearn` compiles correctly on your GPU, you will need to explicitly "pin" your cuDNN
61+
> library version using this command (after installing Cuda-12 JAX):
62+
>
63+
> ```bash
64+
> pip install --force-reinstall "nvidia-cudnn-cu12==9.10.2.21"
65+
> ```
66+
4867
### User Installation
4968
5069
<i>Setup</i>: The easiest way to install ngc-learn is through <code>pip</code>:

0 commit comments

Comments
 (0)