From 4548e63e019d52b366c14c66ad8eafa79eaa6c39 Mon Sep 17 00:00:00 2001
From: "John D. Pope" <jp@bellgeorge.com>
Date: Wed, 3 Feb 2021 06:12:39 +1100
Subject: [PATCH] Update docker_run.sh

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

diff --git a/docker_run.sh b/docker_run.sh
index 361281780..a843b8632 100755
--- a/docker_run.sh
+++ b/docker_run.sh
@@ -29,8 +29,11 @@ rest=$@
 IMAGE="${IMAGE:-sg2ada:latest}"
 
 CONTAINER_ID=$(docker inspect --format="{{.Id}}" ${IMAGE} 2> /dev/null)
+WORKDIR=$(pwd)
 if [[ "${CONTAINER_ID}" ]]; then
-    docker run --gpus all -it --rm -v `pwd`:/scratch --user $(id -u):$(id -g) \
+    docker run --shm-size=1g --ulimit memlock=-1 --ulimit stack=67108864 --gpus all \
+    --mount type=bind,source="$WORKDIR",target=/workspace \
+    -it --rm -v `pwd`:/scratch --user $(id -u):$(id -g) \
         --workdir=/scratch -e HOME=/scratch $IMAGE $@
 else
     echo "Unknown container image: ${IMAGE}"