@@ -8,16 +8,16 @@ ARG OPEN_MPI_VERSION=4.0.1
8
8
9
9
# Python won’t try to write .pyc or .pyo files on the import of source modules
10
10
# Force stdin, stdout and stderr to be totally unbuffered. Good for logging
11
- ENV PYTHONDONTWRITEBYTECODE=1
12
- ENV PYTHONUNBUFFERED=1
13
- ENV PYTHONIOENCODING=UTF-8
14
- ENV LANG=C.UTF-8
15
- ENV LC_ALL=C.UTF-8
16
- ENV LD_LIBRARY_PATH="${LD_LIBRARY_PATH}:/usr/local/lib"
17
- ENV LD_LIBRARY_PATH="${LD_LIBRARY_PATH}:/opt/conda/lib"
18
- ENV PATH=/opt/conda/bin:$PATH
19
- ENV SAGEMAKER_TRAINING_MODULE=sagemaker_pytorch_container.training:main
20
- ENV DGLBACKEND=pytorch
11
+ ENV PYTHONDONTWRITEBYTECODE=1
12
+ ENV PYTHONUNBUFFERED=1
13
+ ENV PYTHONIOENCODING=UTF-8
14
+ ENV LANG=C.UTF-8
15
+ ENV LC_ALL=C.UTF-8
16
+ ENV LD_LIBRARY_PATH="${LD_LIBRARY_PATH}:/usr/local/lib"
17
+ ENV LD_LIBRARY_PATH="${LD_LIBRARY_PATH}:/opt/conda/lib"
18
+ ENV PATH=/opt/conda/bin:$PATH
19
+ ENV SAGEMAKER_TRAINING_MODULE=sagemaker_pytorch_container.training:main
20
+ ENV DGLBACKEND=pytorch
21
21
22
22
WORKDIR /
23
23
@@ -52,8 +52,8 @@ RUN wget https://www.open-mpi.org/software/ompi/v4.0/downloads/openmpi-$OPEN_MPI
52
52
# The ENV variables declared below are changed in the previous section
53
53
# Grouping these ENV variables in the first section causes
54
54
# ompi_info to fail. This is only observed in CPU containers
55
- ENV PATH="$PATH:/home/.openmpi/bin"
56
- ENV LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/home/.openmpi/lib/"
55
+ ENV PATH="$PATH:/home/.openmpi/bin"
56
+ ENV LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/home/.openmpi/lib/"
57
57
RUN ompi_info --parsable --all | grep mpi_built_with_cuda_support:value
58
58
59
59
# Copy workaround script for incorrect hostname
@@ -84,7 +84,7 @@ RUN curl -o ~/miniconda.sh -O https://repo.continuum.io/miniconda/Miniconda3-la
84
84
requests==2.22.0 \
85
85
&& conda install -c dglteam -y dgl==0.4.1 \
86
86
&& /opt/conda/bin/conda clean -ya \
87
- && conda clean -ya
87
+ && conda clean -ya
88
88
89
89
# The following section uninstalls torch and torchvision before installing the
90
90
# custom versions from an S3 bucket. This will need to be removed in the future
@@ -95,6 +95,7 @@ RUN pip install --upgrade pip --trusted-host pypi.org --trusted-host files.pytho
95
95
fastai==1.0.59 \
96
96
scipy==1.2.2 \
97
97
smdebug==0.5.0.post0 \
98
+ sagemaker==1.50.17 \
98
99
sagemaker-experiments==0.1.3 \
99
100
/sagemaker_pytorch_training.tar.gz \
100
101
&& pip install --no-cache-dir -U https://pytorch-aws.s3.amazonaws.com/pytorch-1.4.0/py3/cpu/torch-1.4.0-cp36-cp36m-manylinux1_x86_64.whl \
0 commit comments