Răsfoiți Sursa

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 luni în urmă
părinte
comite
f2b103be2e
1 a modificat fișierele cu 1 adăugiri și 1 ștergeri
  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)"
 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)"
   if [ -z "${GPU_SELECT}" ]; then
     echo "No NVIDIA GPUs detected or NVIDIA Container Toolkit not configured. Exiting."