diff --git a/docker_run.sh b/docker_run.sh index 341c6cfe1e7b9f30d61b542fe8928e96bb34337d..23dd1a6edcf9fd139aff8f5461b637818020e5cd 100755 --- a/docker_run.sh +++ b/docker_run.sh @@ -35,6 +35,7 @@ docker run \ --gpus all \ -v "$(pwd)":/submission \ -w /submission \ + --shm-size=10.24gb\ $IMAGE_NAME python local_evaluation.py # Note: We assume you have nvidia-container-toolkit installed and configured