Merge e77c71b3f8 into 6c3faafed2
This commit is contained in:
commit
44251103f1
|
|
@ -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
|
||||
|
|
|
|||
Loading…
Reference in New Issue