@@ -87,11 +87,20 @@ python3 -m pip install --upgrade build
8787python3 -m pip install --upgrade packaging
8888python3 -m pip install --upgrade wheel
8989python3 -m pip install --upgrade setuptools
90- # cupy and h5py need an older Cython
90+ # cupy needs an older Cython
9191# https://github.com/cupy/cupy/issues/4610
92- # https://github.com/h5py/h5py/issues/2268
9392python3 -m pip install --upgrade " cython<3.0"
93+ # cupy for ROCm
94+ # https://docs.cupy.dev/en/stable/install.html#building-cupy-for-rocm-from-source
95+ # https://github.com/cupy/cupy/issues/7830
96+ CC=cc CXX=CC \
97+ CUPY_INSTALL_USE_HIP=1 \
98+ ROCM_HOME=${ROCM_PATH} \
99+ HCC_AMDGPU_TARGET=${AMREX_AMD_ARCH} \
100+ python3 -m pip install -v cupy
101+ python3 -m pip install --upgrade " cython>=3.0" # for latest mpi4py and everything else
94102python3 -m pip install --upgrade numpy
103+ python3 -m pip install --upgrade h5py
95104python3 -m pip install --upgrade pandas
96105python3 -m pip install --upgrade scipy
97106MPICC=" cc -shared" python3 -m pip install --upgrade mpi4py --no-cache-dir --no-build-isolation --no-binary mpi4py
@@ -100,14 +109,6 @@ python3 -m pip install --upgrade matplotlib
100109python3 -m pip install --upgrade yt
101110# install or update WarpX dependencies such as picmistandard
102111python3 -m pip install --upgrade -r $HOME /src/warpx/requirements.txt
103- # cupy for ROCm
104- # https://docs.cupy.dev/en/stable/install.html#building-cupy-for-rocm-from-source
105- # https://github.com/cupy/cupy/issues/7830
106- CC=cc CXX=CC \
107- CUPY_INSTALL_USE_HIP=1 \
108- ROCM_HOME=${ROCM_PATH} \
109- HCC_AMDGPU_TARGET=${AMREX_AMD_ARCH} \
110- python3 -m pip install -v cupy
111112# optional: for optimas (based on libEnsemble & ax->botorch->gpytorch->pytorch)
112113# python3 -m pip install --upgrade torch --index-url https://download.pytorch.org/whl/rocm5.4.2
113114# python3 -m pip install -r $HOME/src/warpx/Tools/optimas/requirements.txt
0 commit comments