@@ -87,11 +87,20 @@ python3 -m pip install --upgrade build
87
87
python3 -m pip install --upgrade packaging
88
88
python3 -m pip install --upgrade wheel
89
89
python3 -m pip install --upgrade setuptools
90
- # cupy and h5py need an older Cython
90
+ # cupy needs an older Cython
91
91
# https://github.com/cupy/cupy/issues/4610
92
- # https://github.com/h5py/h5py/issues/2268
93
92
python3 -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
94
102
python3 -m pip install --upgrade numpy
103
+ python3 -m pip install --upgrade h5py
95
104
python3 -m pip install --upgrade pandas
96
105
python3 -m pip install --upgrade scipy
97
106
MPICC=" 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
100
109
python3 -m pip install --upgrade yt
101
110
# install or update WarpX dependencies such as picmistandard
102
111
python3 -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
111
112
# optional: for optimas (based on libEnsemble & ax->botorch->gpytorch->pytorch)
112
113
# python3 -m pip install --upgrade torch --index-url https://download.pytorch.org/whl/rocm5.4.2
113
114
# python3 -m pip install -r $HOME/src/warpx/Tools/optimas/requirements.txt
0 commit comments