diff --git a/scripts/install.sh b/scripts/install.sh index 9c232400f..ddc0b0a47 100644 --- a/scripts/install.sh +++ b/scripts/install.sh @@ -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 @@ -239,7 +244,7 @@ CUDA_REPO_ERR_MSG="NVIDIA GPU detected, but your OS and Architecture are not sup # ref: https://docs.nvidia.com/cuda/cuda-installation-guide-linux/index.html#fedora install_cuda_driver_yum() { status 'Installing NVIDIA repository...' - + case $PACKAGE_MANAGER in yum) $SUDO $PACKAGE_MANAGER -y install yum-utils