Allowing arguments for deep_learning_container.py for tf1.15 (#330)

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

Author: TusharKanekiDey

docker/1.15.2/py2/Dockerfile.cpu

@@ -113,7 +113,7 @@ 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
+ADD https://raw.githubusercontent.com/aws/aws-deep-learning-containers-utils/master/deep_learning_container.py /usr/local/bin/deep_learning_container.py
 
 RUN chmod +x /usr/local/bin/deep_learning_container.py
 

docker/1.15.2/py2/Dockerfile.gpu

@@ -155,7 +155,7 @@ 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
+ADD https://raw.githubusercontent.com/aws/aws-deep-learning-containers-utils/master/deep_learning_container.py /usr/local/bin/deep_learning_container.py
 
 RUN chmod +x /usr/local/bin/deep_learning_container.py
 

docker/1.15.2/py3/Dockerfile.cpu

@@ -116,7 +116,7 @@ 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
+ADD https://raw.githubusercontent.com/aws/aws-deep-learning-containers-utils/master/deep_learning_container.py /usr/local/bin/deep_learning_container.py
 
 RUN chmod +x /usr/local/bin/deep_learning_container.py
 

docker/1.15.2/py3/Dockerfile.gpu

@@ -162,7 +162,7 @@ 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
+ADD https://raw.githubusercontent.com/aws/aws-deep-learning-containers-utils/master/deep_learning_container.py /usr/local/bin/deep_learning_container.py
 
 RUN chmod +x /usr/local/bin/deep_learning_container.py