Merge pull request #1240 from IuliuNovac/master

FIX missing GPU in wsl
pull/1251/head
Piero Toffanin 2022-09-21 00:10:28 -04:00 zatwierdzone przez GitHub
commit 29ac754cc1
Nie znaleziono w bazie danych klucza dla tego podpisu
ID klucza GPG: 4AEE18F83AFDEB23
1 zmienionych plików z 8 dodań i 0 usunięć

Wyświetl plik

@ -185,6 +185,14 @@ detect_gpus(){
set -e
return
fi
NVIDIA_DEVICE=$(nvidia-smi -L | grep "GPU 0:" | awk -F ': ' '{print $2}' | awk -F '(' '{print $1}')
#IF the NVIDIA_DEVICE has the word NVIDIA in it, then enable NVIDIA
if [[ $NVIDIA_DEVICE == *"NVIDIA"* ]]; then
echo "GPU_NVIDIA has been found"
export GPU_NVIDIA=true
set -e
return
fi
if lspci | grep "VGA.*Intel"; then
echo "GPU_INTEL has been found"