Просмотр исходного кода

vast.ai fix | "No devices were found" fallback for GPU_SELECT (#73)

* "No devices were found" fallback for GPU_SELECT

* Minor edits

---------

Co-authored-by: Seungmin Kim <8457324+ehfd@users.noreply.github.com>
DanielV74 10 месяцев назад
Родитель
Сommit
f2b103be2e
1 измененных файлов с 1 добавлено и 1 удалено
  1. 1 1
      entrypoint.sh

+ 1 - 1
entrypoint.sh

@@ -94,7 +94,7 @@ else
   export GPU_SELECT="$(nvidia-smi --query-gpu=uuid --format=csv,noheader | head -n1)"
   export GPU_SELECT="$(nvidia-smi --query-gpu=uuid --format=csv,noheader | head -n1)"
 fi
 fi
 
 
-if [ -z "${GPU_SELECT}" ]; then
+if [ -z "${GPU_SELECT}" ] || [ "${GPU_SELECT#*No devices}" != "${GPU_SELECT}" ]; then
   export GPU_SELECT="$(nvidia-smi --query-gpu=uuid --format=csv,noheader | head -n1)"
   export GPU_SELECT="$(nvidia-smi --query-gpu=uuid --format=csv,noheader | head -n1)"
   if [ -z "${GPU_SELECT}" ]; then
   if [ -z "${GPU_SELECT}" ]; then
     echo "No NVIDIA GPUs detected or NVIDIA Container Toolkit not configured. Exiting."
     echo "No NVIDIA GPUs detected or NVIDIA Container Toolkit not configured. Exiting."