File tree Expand file tree Collapse file tree 1 file changed +5
-2
lines changed Expand file tree Collapse file tree 1 file changed +5
-2
lines changed Original file line number Diff line number Diff line change @@ -67,7 +67,6 @@ RUN conda install -c \
67
67
&& conda install -y \
68
68
scikit-learn==0.21.2 \
69
69
pandas==0.25.0 \
70
- pillow==6.2.1 \
71
70
h5py==2.9.0 \
72
71
requests==2.22.0 \
73
72
&& conda clean -ya \
@@ -78,7 +77,8 @@ RUN conda install -c \
78
77
&& pip install https://download.pytorch.org/whl/cpu/torchvision-0.4.2%2Bcpu-cp36-cp36m-linux_x86_64.whl \
79
78
&& pip install https://s3.amazonaws.com/amazonei-pytorch/torch_eia-1.3.1-cp36-cp36m-manylinux1_x86_64.whl \
80
79
&& pip install graphviz==$GRAPHVIZ_VERSION \
81
- && pip install mxnet-model-server==$MMS_VERSION
80
+ && pip install mxnet-model-server==$MMS_VERSION \
81
+ && pip install pillow==7.1.0
82
82
83
83
RUN useradd -m model-server \
84
84
&& mkdir -p /home/model-server/tmp \
@@ -93,6 +93,9 @@ RUN pip install --no-cache-dir "sagemaker-pytorch-inference<2"
93
93
94
94
RUN curl https://aws-dlc-licenses.s3.amazonaws.com/pytorch/license.txt -o /license.txt
95
95
96
+ RUN conda install -y -c conda-forge pyyaml==5.3.1 \
97
+ && pip install sagemaker-containers==2.8.6
98
+
96
99
RUN wget https://amazonei-tools.s3.amazonaws.com/v${HEALTH_CHECK_VERSION}/ei_tools_${HEALTH_CHECK_VERSION}.tar.gz -O /opt/ei_tools_${HEALTH_CHECK_VERSION}.tar.gz \
97
100
&& tar -xvf /opt/ei_tools_${HEALTH_CHECK_VERSION}.tar.gz -C /opt/ \
98
101
&& rm -rf /opt/ei_tools_${HEALTH_CHECK_VERSION}.tar.gz \
You can’t perform that action at this time.
0 commit comments