build gpu docker image on linux if nvidia-smi exists

beam-cloud · Jan 22, 2024 · 20336e3 · 20336e3
1 parent b653d89
commit 20336e3
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/bin/k3d.sh b/bin/k3d.sh
@@ -16,8 +16,12 @@ k3d_up() {
   os_type="$(uname)"
   case "$os_type" in
     Linux*)
-      check_gpu_linux && extra_args="--gpus=all --image=localhost:5001/rancher/k3s:latest" || extra_args=""
-      docker build . -f ./docker/Dockerfile.k3d -t localhost:5001/rancher/k3s:latest
+      if check_gpu_linux; do
+        extra_args="--gpus=all --image=localhost:5001/rancher/k3s:latest"
+        docker build . -f ./docker/Dockerfile.k3d -t localhost:5001/rancher/k3s:latest
+      elif
+        extra_args=""
+      fi
       ;;
     Darwin*)
       extra_args=""