2# Install CuPy and the CUDA 12 runtime libraries as pip wheels, for machines
3# with an NVIDIA driver but no system CUDA toolkit. Then source scripts/gpuenv.sh.
4set -euo pipefail
5python3 -m pip install --user cupy-cuda12x numpy matplotlib \
6 nvidia-cuda-runtime-cu12 nvidia-cuda-nvrtc-cu12 nvidia-cublas-cu12 nvidia-cusparse-cu12 \
7 nvidia-cufft-cu12 nvidia-curand-cu12 nvidia-cusolver-cu12 nvidia-nvjitlink-cu12
8echo "Now run: source scripts/gpuenv.sh"