Adding of deep_learning_container.py in Tf1.15 (#327)

Co-authored-by: Tushar Dey <[email protected]>

Author: TusharKanekiDey

docker/1.15.2/py2/Dockerfile.cpu

@@ -113,6 +113,10 @@ RUN pip install --no-cache-dir -U \
  && pip install --no-cache-dir -U \
     horovod==0.18.2
 
+COPY deep_learning_container.py /usr/local/bin/deep_learning_container.py
+
+RUN chmod +x /usr/local/bin/deep_learning_container.py
+
 RUN curl https://aws-dlc-licenses.s3.amazonaws.com/tensorflow/license.txt -o /license.txt
 
 CMD ["bin/bash"]

docker/1.15.2/py2/Dockerfile.gpu

@@ -155,6 +155,10 @@ RUN cat /etc/ssh/ssh_config | grep -v StrictHostKeyChecking > /etc/ssh/ssh_confi
  && echo "    StrictHostKeyChecking no" >> /etc/ssh/ssh_config.new \
  && mv /etc/ssh/ssh_config.new /etc/ssh/ssh_config
 
+COPY deep_learning_container.py /usr/local/bin/deep_learning_container.py
+
+RUN chmod +x /usr/local/bin/deep_learning_container.py
+
 RUN curl https://aws-dlc-licenses.s3.amazonaws.com/tensorflow/license.txt -o /license.txt
 
 CMD ["bin/bash"]

docker/1.15.2/py3/Dockerfile.cpu

@@ -116,6 +116,10 @@ RUN pip install --no-cache-dir -U \
     awscli\
  && rm -f $FRAMEWORK_SUPPORT_INSTALLABLE
 
+COPY deep_learning_container.py /usr/local/bin/deep_learning_container.py
+
+RUN chmod +x /usr/local/bin/deep_learning_container.py
+
 RUN curl https://aws-dlc-licenses.s3.amazonaws.com/tensorflow/license.txt -o /license.txt
 
 CMD ["bin/bash"]

docker/1.15.2/py3/Dockerfile.gpu

@@ -162,6 +162,10 @@ RUN cat /etc/ssh/ssh_config | grep -v StrictHostKeyChecking > /etc/ssh/ssh_confi
  && echo "    StrictHostKeyChecking no" >> /etc/ssh/ssh_config.new \
  && mv /etc/ssh/ssh_config.new /etc/ssh/ssh_config
 
+COPY deep_learning_container.py /usr/local/bin/deep_learning_container.py
+
+RUN chmod +x /usr/local/bin/deep_learning_container.py
+
 RUN curl https://aws-dlc-licenses.s3.amazonaws.com/tensorflow/license.txt -o /license.txt
 
 CMD ["bin/bash"]