Fixed multinode entrypoint to exit properly (#186)

Signed-off-by: tylertitsworth <[email protected]>
intel · Jun 28, 2024 · 5fcb25c · 5fcb25c
1 parent 8ea038a
commit 5fcb25c
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/pytorch/Dockerfile b/pytorch/Dockerfile
@@ -123,8 +123,7 @@ set -e \n\
 set -a \n\
 source ~/.startup \n\
 set +a \n\
-eval \"\$@\" \n\
-tail -f /dev/null" >> /usr/local/bin/dockerd-entrypoint.sh && \
+eval \"\$@\"" >> /usr/local/bin/dockerd-entrypoint.sh && \
     chmod +x /usr/local/bin/dockerd-entrypoint.sh
 
 RUN echo 'HostKey /etc/ssh/ssh_host_dsa_key' > /var/run/sshd_config && \