From 2cb41121458e69085418830027a6f875b293ca38 Mon Sep 17 00:00:00 2001 From: Samuel GIFFARD Date: Thu, 25 Jan 2024 18:09:53 +0100 Subject: [PATCH] Enhance GPU detection logic to support WSL2 environments This commit updates the GPU detection logic in the setup script to accommodate environments where NVIDIA GPUs might not be directly listed by `lspci`, such as in Windows Subsystem for Linux 2 (WSL2). The revised logic now attempts to identify NVIDIA GPUs using both `lspci` and `nvidia-smi`, ensuring broader compatibility and recognizing GPUs that are accessible in WSL2. This change allows the script to correctly detect NVIDIA GPUs in a wider range of setups, improving user experience and script robustness in diverse environments. --- run-compose.sh | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/run-compose.sh b/run-compose.sh index 0557bce95..7d1ff74d0 100755 --- a/run-compose.sh +++ b/run-compose.sh @@ -11,8 +11,8 @@ TICK='\u2713' # Detect GPU driver get_gpu_driver() { - # Detect NVIDIA GPUs - if lspci | grep -i nvidia >/dev/null; then + # Detect NVIDIA GPUs using lspci or nvidia-smi + if lspci | grep -i nvidia >/dev/null || nvidia-smi >/dev/null 2>&1; then echo "nvidia" return fi