From fd5ea7aeb4071e7d96774e2a6d98030eac896fa5 Mon Sep 17 00:00:00 2001
From: Lara CODECA <lara.codeca@gmail.com>
Date: Mon, 10 Aug 2020 15:19:29 +0100
Subject: [PATCH] additional SHM required for Deep RL

---
 docker-cmd-linux.sh | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/docker-cmd-linux.sh b/docker-cmd-linux.sh
index 326bc8b..416df8c 100755
--- a/docker-cmd-linux.sh
+++ b/docker-cmd-linux.sh
@@ -28,6 +28,7 @@ LEARN_DIR=""
 COMMAND=""
 EXP=""
 DETACH=false
+SHM_SIZE="10g"
 
 function print_help {
     echo "Parameters:"
@@ -43,6 +44,7 @@ function print_help {
     echo "  DETACH: ($DETACH) [--detach]"
     echo "  DEVELOPMENT dir \"$DEVEL_DIR\" [-d, --devel]"
     echo "  LEARNING dir \"$LEARN_DIR\" [-l, --learn]"
+    echo "  SHM_SIZE \"$SHM_SIZE\" [--shm-size]"
 }
 
 for arg in "$@"
@@ -93,6 +95,9 @@ do
         -l=*|--learn=*)
         LEARN_DIR="${arg#*=}"
         ;;
+        --shm-size=*)
+        SHM_SIZE="${arg#*=}"
+        ;;
         *)
         # unknown option
         echo "Unknown option \"$arg\""
@@ -149,7 +154,7 @@ if [[ "$RUN" = true ]]; then
             --volume $XAUTHORITY:/home/alice/.Xauthority \
             --volume /tmp/.X11-unix:/tmp/.X11-unix \
             --privileged $MOUNT_DEVEL $MOUNT_LEARN \
-            --shm-size 256m $GPU_OPT $CONT_NAME \
+            --shm-size $SHM_SIZE $GPU_OPT $CONT_NAME \
             -it $DETACH --rm $IMAGE_NAME:latest"
     echo "$OPTIRUN_OPT docker run $RUN_OPT $COMMAND"