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

Signed-off-by: red-hat-konflux <126015336+red-hat-konflux[bot]@users.noreply.github.com>
red-hat-data-services · Jul 23, 2024 · b94ab4c · b94ab4c
1 parent 78eedd8
commit b94ab4c
Showing 1 changed file with 1 addition and 1 deletion.
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.06-py3
 
 # Set the working directory in the container
 WORKDIR /app