-
Notifications
You must be signed in to change notification settings - Fork 72
PyTorch 1.3.1 #13
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
PyTorch 1.3.1 #13
Conversation
AWS CodeBuild CI Report
Powered by github-codebuild-logs, available on the AWS Serverless Application Repository |
docker/1.3.0/py3/Dockerfile.gpu
Outdated
|
||
ARG PYTORCH_VERSION=1.3.0 | ||
ARG TORCHVISION_VERSION=0.4.1 | ||
ARG MMS_VERSION=1.0.7 |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Please update this to the latest 1.0.8 everywhere.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
done
docker/1.3.0/py3/Dockerfile.gpu
Outdated
/opt/conda/bin/conda clean -ya | ||
ENV PATH /opt/conda/bin:$PATH | ||
|
||
ARG PYTORCH_VERSION=1.3.0 |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Please update to PyTorch 1.3.1 everywhere as discussed, thank you.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
done
AWS CodeBuild CI Report
Powered by github-codebuild-logs, available on the AWS Serverless Application Repository |
docker/1.3.1/py2/Dockerfile.cpu
Outdated
curl \ | ||
git \ | ||
wget \ | ||
vim \ |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Please sort apt-get libraries
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
done
docker/1.3.1/py2/Dockerfile.cpu
Outdated
|
||
# See http://bugs.python.org/issue19846 | ||
ENV LANG C.UTF-8 | ||
|
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Move all ARGs and ENVs to the top. ARGs before ENV. Open MP envs are an exception.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
done
docker/1.3.1/py2/Dockerfile.cpu
Outdated
# Add arguments to achieve the version, python and url | ||
ARG PYTHON_VERSION=2.7 | ||
|
||
RUN curl -o ~/miniconda.sh -O https://repo.continuum.io/miniconda/Miniconda3-latest-Linux-x86_64.sh && \ |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Use && in the beginning of the next line (applies to all multi-line RUNs in all docker files)
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
done
docker/1.3.1/py2/Dockerfile.cpu
Outdated
mkl==2019.4 \ | ||
mkl-include==2019.4 \ | ||
cython==0.29.12 \ | ||
typing==3.7.4 && \ |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Update all libraries to latest versions
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Easiest way to do this is to unpin all versions, run the build, if it's successful, pin versions according to the ones that come up in a pip freeze | grep <package>
within the container.
docker/1.3.1/py2/Dockerfile.cpu
Outdated
pillow==6.1.0 \ | ||
h5py==2.9.0 \ | ||
requests==2.22.0 && \ | ||
conda install pytorch==$PYTORCH_VERSION torchvision==$TORCHVISION_VERSION cpuonly -c pytorch && \ |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Align according to best practices doc
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
done
docker/1.3.1/py2/Dockerfile.gpu
Outdated
@@ -0,0 +1,101 @@ | |||
FROM nvidia/cuda:10.0-cudnn7-devel-ubuntu16.04 |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Please make same changes as CPU to GPU.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
done
docker/1.3.1/py2/Dockerfile.gpu
Outdated
ARG PYTHON_VERSION=2.7 | ||
|
||
# Install OpenSSH, Allow OpenSSH to talk to containers without asking for confirmation | ||
RUN apt-get install -y --no-install-recommends openssh-client openssh-server && \ |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Move each install to one line.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
done
docker/1.3.1/py2/mms-entrypoint.py
Outdated
import sys | ||
import os.path | ||
|
||
from sagemaker_pytorch_serving_container import serving |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Should this be moved inside the if condition? Keep this consistent with the mms-entrypoint for mxnet.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
@@ -0,0 +1,81 @@ | |||
FROM ubuntu:16.04 |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Same as previous docker files.
docker/1.3.1/py3/Dockerfile.gpu
Outdated
@@ -0,0 +1,93 @@ | |||
FROM nvidia/cuda:10.0-cudnn7-devel-ubuntu16.04 |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Same as previous docker file.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
done
AWS CodeBuild CI Report
Powered by github-codebuild-logs, available on the AWS Serverless Application Repository |
AWS CodeBuild CI Report
Powered by github-codebuild-logs, available on the AWS Serverless Application Repository |
AWS CodeBuild CI Report
Powered by github-codebuild-logs, available on the AWS Serverless Application Repository |
docker/1.2.0/py2/Dockerfile.cpu
Outdated
@@ -50,6 +50,7 @@ RUN conda install -c conda-forge awscli==1.16.210 opencv==4.0.1 && \ | |||
pillow==6.1.0 \ | |||
h5py==2.9.0 \ | |||
requests==2.22.0 && \ | |||
|
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Remove empty line
docker/1.2.0/py2/Dockerfile.cpu
Outdated
@@ -50,6 +50,7 @@ RUN conda install -c conda-forge awscli==1.16.210 opencv==4.0.1 && \ | |||
pillow==6.1.0 \ | |||
h5py==2.9.0 \ | |||
requests==2.22.0 && \ |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
&& should be in the next line. This needs to be fixed across the file.
docker/1.2.0/py2/Dockerfile.cpu
Outdated
@@ -50,6 +50,7 @@ RUN conda install -c conda-forge awscli==1.16.210 opencv==4.0.1 && \ | |||
pillow==6.1.0 \ | |||
h5py==2.9.0 \ |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Alignment changes required. The names of the libraries must be aligned with conda...
docker/1.2.0/py2/Dockerfile.cpu
Outdated
@@ -50,6 +50,7 @@ RUN conda install -c conda-forge awscli==1.16.210 opencv==4.0.1 && \ | |||
pillow==6.1.0 \ |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Lines 28~41 alignment is off by one space.
docker/1.3.1/py2/mms-entrypoint.py
Outdated
import sys | ||
import os.path | ||
|
||
from sagemaker_pytorch_serving_container import serving |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
pillow==5.4.1 \ | ||
h5py==2.9.0 \ | ||
requests==2.22.0 \ | ||
&& conda install -c \ |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
You can combine all commands of '-c pytorch' together i.e. magma-cuda100 is also being fetched from pytorch channel.
AWS CodeBuild CI Report
Powered by github-codebuild-logs, available on the AWS Serverless Application Repository |
Added PyTorch 1.3.1 support.
By submitting this pull request, I confirm that you can use, modify, copy, and redistribute this contribution, under the terms of your choice.