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