diff options
| author |  Axel Huebl <axel.huebl@plasma.ninja>
| 2023-09-12 16:14:17 -0700 |
|---|---|---|
| committer |  GitHub <noreply@github.com>
| 2023-09-12 16:14:17 -0700 |
| commit | 51f908bf15e737ca4ac0cc2871de2d2de77e43a7 (patch) | |
| tree | 15eaa73eeae0f16c9219d2ffc4c3acbe714b06f8 | |
| parent | 12a695d9fa518ad23a4f0375c0ccbafcabb80473 (diff) | |
| download | WarpX-51f908bf15e737ca4ac0cc2871de2d2de77e43a7.tar.gz WarpX-51f908bf15e737ca4ac0cc2871de2d2de77e43a7.tar.zst WarpX-51f908bf15e737ca4ac0cc2871de2d2de77e43a7.zip | |
Doc: cupy on Perlmutter (NERSC) (#4289)
Document & automate how to install `cupy` on Perlmutter.
| -rwxr-xr-x | Tools/machines/perlmutter-nersc/install_gpu_dependencies.sh | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/Tools/machines/perlmutter-nersc/install_gpu_dependencies.sh b/Tools/machines/perlmutter-nersc/install_gpu_dependencies.sh index a60c3ba75..7bcc70539 100755 --- a/Tools/machines/perlmutter-nersc/install_gpu_dependencies.sh +++ b/Tools/machines/perlmutter-nersc/install_gpu_dependencies.sh @@ -127,6 +127,7 @@ python3 -m pip install --upgrade matplotlib python3 -m pip install --upgrade yt # install or update WarpX dependencies such as picmistandard python3 -m pip install --upgrade -r $HOME/src/warpx/requirements.txt +python3 -m pip install cupy-cuda11x # CUDA 11.7 compatible wheel # optional: for libEnsemble python3 -m pip install -r $HOME/src/warpx/Tools/LibEnsemble/requirements.txt # optional: for optimas (based on libEnsemble & ax->botorch->gpytorch->pytorch) |