[run_docker] Update gpu flag

larics · Jan 4, 2024 · 32249cd · 32249cd
1 parent d5e1d3a
commit 32249cd
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/run_docker.sh b/run_docker.sh
@@ -51,13 +51,16 @@ fi
 
 # Check if using GPU
 gpu_enabled="--gpus all"
+if [ "$distro" == "focal-nogpu" ]; then
+  gpu_enabled=""
+fi
+run_args="$gpu_enabled $run_args"
 
 docker run \
   $run_args \
   -it \
   --network host \
   --privileged \
-  $gpu_enabled \
   --volume=$XSOCK:$XSOCK:rw \
   --volume=$XAUTH:$XAUTH:rw \
   --env="XAUTHORITY=${XAUTH}" \