From 32249cd08453c3bd0e769df22eefc4e3dfa7cad5 Mon Sep 17 00:00:00 2001
From: Lovro Markovic <lovro.markovic@fer.hr>
Date: Thu, 4 Jan 2024 13:07:58 +0100
Subject: [PATCH] [run_docker] Update gpu flag

---
 run_docker.sh | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/run_docker.sh b/run_docker.sh
index 1377b80..1a3df00 100755
--- 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}" \