fix: nvidia-smi path issue for root user in WSL

pull/1513/head
Jiangzhou Li 11 months ago committed by Lawrence Li
parent eefe77490a
commit 0c85b98fe3

@ -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

@ -1 +1 @@
aa6ebe9416c158f756e6efa0741f2d9bee337d4b2e6e6fde92852c6d624e94eb install.sh
692c0927451c841b7a4963e457788c37fc09a0a414889e61c3ea7d84cd8b90b4 install.sh

Loading…
Cancel
Save