From a647c1032d43eee0036724cf3fe10cbed9191bba Mon Sep 17 00:00:00 2001 From: daniel-dudt Date: Tue, 17 Dec 2024 11:19:29 -0700 Subject: [PATCH] update Perlmutter install docs --- docs/installation.rst | 23 ++++++----------------- 1 file changed, 6 insertions(+), 17 deletions(-) diff --git a/docs/installation.rst b/docs/installation.rst index 9df67563b..403b1c658 100644 --- a/docs/installation.rst +++ b/docs/installation.rst @@ -96,17 +96,18 @@ specific JAX GPU installation instructions, as that is the main installation dif **Note that DESC does not always test on or guarantee support of the latest version of JAX (which does not have a stable 1.0 release yet), and thus older versions of GPU-accelerated versions of JAX may need to be installed, which may in turn require lower versions of JaxLib, as well as CUDA and CuDNN.** + Perlmutter (NERSC) ++++++++++++++++++++++++++++++ -These instructions were tested and confirmed to work on the Perlmutter supercomputer at NERSC on June 18, 2024. +These instructions were tested and confirmed to work on the Perlmutter supercomputer at NERSC on December 17, 2024. Set up the correct cuda environment for jax installation .. code-block:: sh - module load cudatoolkit/12.2 + module load cudatoolkit/12.4 module load cudnn/8.9.3_cuda12 - module load python + module load python/3.11 Check that you have loaded these modules @@ -118,21 +119,9 @@ Create a conda environment for DESC (`following these instructions = 1.7.0, < 2.0.0 - -to - -.. code-block:: sh - - scipy >= 1.7.0, <= 1.11.3 + pip install --upgrade "jax[cuda12]" Clone and install DESC