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