diff --git a/pytorch/Dockerfile b/pytorch/Dockerfile
index c02ca9b26..395916ca2 100644
--- 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 && \