use default nvidia-smi as fallback when on WSL2

This commit is contained in:
Pablo Castellazzi 2026-01-02 05:56:23 -03:00
parent 18fdcc94e5
commit e77c71b3f8
No known key found for this signature in database
1 changed files with 6 additions and 1 deletions

View File

@ -173,6 +173,11 @@ fi
# WSL2 only supports GPUs via nvidia passthrough
# so check for nvidia-smi to determine if GPU is available
if [ "$IS_WSL2" = true ]; then
# WSL2 has nvidia-smi installed by default,
# we add its folder to the end of the PATH as a fallback.
if [ -x /usr/lib/wsl/lib/nvidia-smi ]; then
export PATH=${PATH}:/usr/lib/wsl/lib
fi
if available nvidia-smi && [ -n "$(nvidia-smi | grep -o "CUDA Version: [0-9]*\.[0-9]*")" ]; then
status "Nvidia GPU detected."
fi