diff --git a/pytorch/inference/docker/1.8/py3/Dockerfile.cpu b/pytorch/inference/docker/1.8/py3/Dockerfile.cpu index fe4b598d7364..dae8151a310b 100644 --- a/pytorch/inference/docker/1.8/py3/Dockerfile.cpu +++ b/pytorch/inference/docker/1.8/py3/Dockerfile.cpu @@ -106,7 +106,7 @@ RUN pip uninstall -y torch \ && pip install --no-deps --no-cache-dir -U $PT_TORCHVISION_URL RUN conda install -y -c conda-forge "pyyaml>=5.4,<5.5" -RUN pip install pillow==7.2.0 "awscli<2" +RUN pip install pillow==8.2.0 "awscli<2" RUN pip install --no-cache-dir "sagemaker-pytorch-inference>=2" RUN pip uninstall -y model-archiver multi-model-server \ diff --git a/pytorch/inference/docker/1.8/py3/cu111/Dockerfile.gpu b/pytorch/inference/docker/1.8/py3/cu111/Dockerfile.gpu index e76eea448ee4..c8b71abe8cda 100644 --- a/pytorch/inference/docker/1.8/py3/cu111/Dockerfile.gpu +++ b/pytorch/inference/docker/1.8/py3/cu111/Dockerfile.gpu @@ -127,7 +127,7 @@ RUN pip uninstall -y torch \ && pip install --no-deps --no-cache-dir -U $PT_TORCHVISION_URL RUN conda install -y -c conda-forge "pyyaml>=5.4,<5.5" -RUN pip install pillow==7.2.0 "awscli<2" +RUN pip install pillow==8.2.0 "awscli<2" RUN pip install --no-cache-dir "sagemaker-pytorch-inference>=2"