@@ -45,6 +45,25 @@ Matplotlib (>=3.8.0) and imageio (>=2.31.5) and both plotting and density estima
4545tools (routines within `` ngclearn.utils.density `` ) will require Scikit-learn (>=0.24.2).
4646Many 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