[graviton] update PyTorch 2.1 buildspec to use the latest torchserve (#…

…3411)
aws · Oct 20, 2023 · da48e48 · da48e48
1 parent 119d075
commit da48e48
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/pytorch/inference/buildspec-graviton.yml b/pytorch/inference/buildspec-graviton.yml
@@ -34,7 +34,7 @@ images:
     image_size_baseline: 10000
     device_type: &DEVICE_TYPE cpu
     os_version: &OS_VERSION ubuntu20.04
-    torch_serve_version: &TORCHSERVE_VERSION 0.8.2
+    torch_serve_version: &TORCHSERVE_VERSION 0.9.0
     python_version: &DOCKER_PYTHON_VERSION py3
     tag_python_version: &TAG_PYTHON_VERSION py310
     tag: !join [ *VERSION, "-", *DEVICE_TYPE, "-", *TAG_PYTHON_VERSION, "-", *OS_VERSION , "-ec2"]
@@ -48,7 +48,7 @@ images:
     image_size_baseline: 10000
     device_type: &DEVICE_TYPE cpu
     os_version: &OS_VERSION ubuntu20.04
-    torch_serve_version: &TORCHSERVE_VERSION 0.8.2
+    torch_serve_version: &TORCHSERVE_VERSION 0.9.0
     tool_kit_version: &SM_TOOLKIT_VERSION 2.0.16
     python_version: &DOCKER_PYTHON_VERSION py3
     tag_python_version: &TAG_PYTHON_VERSION py310

diff --git a/pytorch/inference/docker/2.1/py3/Dockerfile.graviton.cpu b/pytorch/inference/docker/2.1/py3/Dockerfile.graviton.cpu
@@ -137,7 +137,7 @@ RUN ln -s /opt/conda/bin/pip /usr/local/bin/pip3 \
     "cryptography>=41.0.2" \
     "ipython>=8.10.0,<9.0" \
     "awscli<2" \
-    "urllib3==1.26.17"
+    "urllib3==1.26.18"
 
 # Ensure PyTorch did not get installed from Conda or pip, prior to now
 RUN pip uninstall -y torch torchvision torchaudio torchdata model-archiver multi-model-server