Increase shared memory

Former-commit-id: 3a5f3a0c8c5069e837ade1997df7139f482bc985
This commit is contained in:
milesial 2021-08-19 10:42:39 +02:00 committed by GitHub
parent ce4567c908
commit fdb686fc43

View file

@ -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
```