use OMP_PROC_BIND=close to not tank torch performance during distribu…

…ted training
molmod · Jun 12, 2024 · e739d0b · e739d0b
1 parent f51c51c
commit e739d0b
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -69,6 +69,7 @@ RUN /bin/bash -c -o pipefail \
 RUN echo '#!/bin/bash' >> /opt/entry.sh && \
  echo 'source /opt/venv/bin/activate' >> /opt/entry.sh && \
  echo 'export PLUMED_KERNEL=/usr/local/plumed/lib/libplumedKernel.so' >> /opt/entry.sh && \
+ echo 'export OMP_PROC_BIND=close' >> /opt/entry.sh && \
  echo '"$@"' >> /opt/entry.sh
 RUN chmod +x /opt/entry.sh
 ENTRYPOINT ["/opt/entry.sh"]