fix wsl2 status check for nvidia cards (#4689)

This commit is contained in:
Jeffrey Morgan 2024-05-28 14:49:46 -07:00 committed by GitHub
parent 6d423b383b
commit 45cbfc5aee
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -134,7 +134,7 @@ fi
# WSL2 only supports GPUs via nvidia passthrough # WSL2 only supports GPUs via nvidia passthrough
# so check for nvidia-smi to determine if GPU is available # so check for nvidia-smi to determine if GPU is available
if [ "$IS_WSL2" = true ]; then if [ "$IS_WSL2" = true ]; then
if available nvidia-smi && [ -z "$(nvidia-smi | grep -o "CUDA Version: [0-9]*\.[0-9]*")" ]; then if available nvidia-smi && [ -n "$(nvidia-smi | grep -o "CUDA Version: [0-9]*\.[0-9]*")" ]; then
status "Nvidia GPU detected." status "Nvidia GPU detected."
fi fi
install_success install_success