From fdb686fc43770cb25160ab00874d5a6c8f1286fd Mon Sep 17 00:00:00 2001 From: milesial Date: Thu, 19 Aug 2021 10:42:39 +0200 Subject: [PATCH] Increase shared memory Former-commit-id: 3a5f3a0c8c5069e837ade1997df7139f482bc985 --- README.md | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/README.md b/README.md index b8c0987..1a3b413 100644 --- a/README.md +++ b/README.md @@ -36,7 +36,7 @@ sudo systemctl restart docker ``` 3. [Download and run the image:](https://hub.docker.com/repository/docker/milesial/unet) ```bash -sudo docker run --rm --gpus all -it milesial/unet +sudo docker run --rm --shm-size=8g --ulimit memlock=-1 --gpus all -it milesial/unet ``` 4. Download the data and run training: @@ -60,7 +60,7 @@ A docker image containing the code and the dependencies is available on [DockerH You can download and jump in the container with ([docker >=19.03](https://docs.docker.com/get-docker/)): ```console -docker run -it --rm --gpus all milesial/unet +docker run -it --rm --shm-size=8g --ulimit memlock=-1 --gpus all milesial/unet ```