1# Source this file: points the loader at the CUDA libraries installed as pip wheels
2# by scripts/setup_gpu.sh. Not needed when a CUDA 12 toolkit is installed system wide.
3NV="$(python3 -c 'import nvidia; print(list(nvidia.__path__)[0])' 2>/dev/null || true)"
4if [ -n "$NV" ]; then
5 for d in cuda_nvrtc cuda_runtime cublas cusparse cufft curand cusolver nvjitlink; do
6 LD_LIBRARY_PATH="$NV/$d/lib${LD_LIBRARY_PATH:+:$LD_LIBRARY_PATH}"
7 done
8 export LD_LIBRARY_PATH
9fi
10export CUPY_CACHE_DIR="${CUPY_CACHE_DIR:-$HOME/.cupy/kernel_cache}"