Skip to content

Commit 3b1dff5

Browse files
update cupy installation in Perlmutter dependencies (BLAST-WarpX#4621)
1 parent a2b9e68 commit 3b1dff5

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

Tools/machines/perlmutter-nersc/install_gpu_dependencies.sh

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -130,9 +130,9 @@ python3 -m pip install --upgrade matplotlib
130130
python3 -m pip install --upgrade yt
131131
# install or update WarpX dependencies such as picmistandard
132132
python3 -m pip install --upgrade -r $HOME/src/warpx/requirements.txt
133-
python3 -m pip install cupy-cuda11x # CUDA 11.7 compatible wheel
133+
python3 -m pip install cupy-cuda12x # CUDA 12 compatible wheel
134134
# optional: for libEnsemble
135135
python3 -m pip install -r $HOME/src/warpx/Tools/LibEnsemble/requirements.txt
136136
# optional: for optimas (based on libEnsemble & ax->botorch->gpytorch->pytorch)
137-
python3 -m pip install --upgrade torch # CUDA 11.7 compatible wheel
137+
python3 -m pip install --upgrade torch # CUDA 12 compatible wheel
138138
python3 -m pip install -r $HOME/src/warpx/Tools/optimas/requirements.txt

0 commit comments

Comments
 (0)