Skip to content

GitLab

"...git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "17612de451244c4c169b1498f1333401dfd3106f"
Unverified Commit 45cbfc5a authored by Jeffrey Morgan's avatar Jeffrey Morgan Committed by GitHub
Browse files

fix wsl2 status check for nvidia cards (#4689)

parent 6d423b38
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
scripts/install.sh
View file @ 45cbfc5a
...@@ -134,7 +134,7 @@ fi ...@@ -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
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment