Update nvcr.io/nvidia/pytorch Docker tag to v24.05

red-hat-data-services · Jun 26, 2024 · 0d1770b · 0d1770b
1 parent 78eedd8
commit 0d1770b
Show file tree

Hide file tree

Showing 5 changed files with 5 additions and 5 deletions.
diff --git a/examples/pytorch/elastic/imagenet/Dockerfile b/examples/pytorch/elastic/imagenet/Dockerfile
@@ -1,7 +1,7 @@
 # We need to use the nvcr.io/nvidia/pytorch image as a base image to support both linux/amd64 and linux_arm64 platforms.
 # PyTorch=2.2.0, cuda=12.3.2
 # Ref: https://docs.nvidia.com/deeplearning/frameworks/pytorch-release-notes/rel-24-01.html#rel-24-01
-ARG BASE_IMAGE=nvcr.io/nvidia/pytorch:24.01-py3
+ARG BASE_IMAGE=nvcr.io/nvidia/pytorch:24.05-py3
 FROM $BASE_IMAGE
 
 WORKDIR /workspace

diff --git a/examples/pytorch/mnist/Dockerfile b/examples/pytorch/mnist/Dockerfile
@@ -1,7 +1,7 @@
 # We need to use the nvcr.io/nvidia/pytorch image as a base image to support both linux/amd64 and linux_arm64 platforms.
 # PyTorch=2.2.0, cuda=12.3.2
 # Ref: https://docs.nvidia.com/deeplearning/frameworks/pytorch-release-notes/rel-24-01.html#rel-24-01
-FROM nvcr.io/nvidia/pytorch:24.01-py3
+FROM nvcr.io/nvidia/pytorch:24.05-py3
 
 RUN pip install tensorboardX==2.6.2
 RUN mkdir -p /opt/mnist

diff --git a/examples/pytorch/mnist/Dockerfile-mpi b/examples/pytorch/mnist/Dockerfile-mpi
@@ -1,7 +1,7 @@
 # We need to use the nvcr.io/nvidia/pytorch image as a base image to support both linux/amd64 and linux_arm64 platforms.
 # PyTorch=2.2.0, cuda=12.3.2
 # Ref: https://docs.nvidia.com/deeplearning/frameworks/pytorch-release-notes/rel-24-01.html#rel-24-01
-FROM nvcr.io/nvidia/pytorch:24.01-py3
+FROM nvcr.io/nvidia/pytorch:24.05-py3
 
 RUN pip install tensorboardX==2.6.2
 RUN mkdir -p /opt/mnist

diff --git a/examples/pytorch/smoke-dist/Dockerfile b/examples/pytorch/smoke-dist/Dockerfile
@@ -1,7 +1,7 @@
 # We need to use the nvcr.io/nvidia/pytorch image as a base image to support both linux/amd64 and linux_arm64 platforms.
 # PyTorch=2.2.0, cuda=12.3.2
 # Ref: https://docs.nvidia.com/deeplearning/frameworks/pytorch-release-notes/rel-24-01.html#rel-24-01
-FROM nvcr.io/nvidia/pytorch:24.01-py3
+FROM nvcr.io/nvidia/pytorch:24.05-py3
 
 RUN mkdir -p /opt/mlkube
 COPY examples/pytorch/smoke-dist/dist_sendrecv.py /opt/mlkube/

diff --git a/sdk/python/kubeflow/trainer/Dockerfile b/sdk/python/kubeflow/trainer/Dockerfile
@@ -1,5 +1,5 @@
 # Use an official Pytorch runtime as a parent image
-FROM nvcr.io/nvidia/pytorch:23.10-py3
+FROM nvcr.io/nvidia/pytorch:24.05-py3
 
 # Set the working directory in the container
 WORKDIR /app