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}" \