diff --git a/install.sh b/install.sh index b2a2d5e..186343f 100755 --- a/install.sh +++ b/install.sh @@ -172,6 +172,10 @@ detect_device() { warn "NVIDIA GPU detected but CUDA is not installed. Please install CUDA." fi DEVICE="cuda" + # Create a symlink for nvidia-smi to allow root users in WSL to detect GPU information. + if [ -f "/usr/lib/wsl/lib/nvidia-smi" ] && [ ! -e "/usr/local/bin/nvidia-smi" ]; then + $SUDO ln -s /usr/lib/wsl/lib/nvidia-smi /usr/local/bin/nvidia-smi + fi fi if check_command "mthreads-gmi"; then diff --git a/install.sh.sha256sum b/install.sh.sha256sum index b01f8ce..0751bb7 100644 --- a/install.sh.sha256sum +++ b/install.sh.sha256sum @@ -1 +1 @@ -aa6ebe9416c158f756e6efa0741f2d9bee337d4b2e6e6fde92852c6d624e94eb install.sh +692c0927451c841b7a4963e457788c37fc09a0a414889e61c3ea7d84cd8b90b4 install.sh