From 10c926947857ea60d28fe572c99302d5257939f2 Mon Sep 17 00:00:00 2001
From: Hannes Hansen <mail@hannesh.de>
Date: Wed, 19 Mar 2025 17:10:47 +0100
Subject: [PATCH] replicate latest pytorch and tensorflow images with gpu
 support; fix https://gitlab.cern.ch/mlops/project/-/issues/144

---
 values-k8s.yaml | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/values-k8s.yaml b/values-k8s.yaml
index 48953e9..4837004 100644
--- a/values-k8s.yaml
+++ b/values-k8s.yaml
@@ -309,6 +309,12 @@ items:
       gcr.io/knative-releases/knative.dev/serving/cmd/webhook@sha256:48aee2733721ecc77956abc5a2ca072853a669ebc97519beb48f7b3da8455e67
       gcr.io/ml-pipeline/workflow-controller:v3.4.17-license-compliance
       docker.io/kserve/xgbserver:v0.13.1
+      docker.io/pytorch/pytorch:2.6.0-cuda12.6-cudnn9-runtime
+      docker.io/pytorch/pytorch:2.6.0-cuda12.4-cudnn9-runtime
+      docker.io/pytorch/pytorch:2.6.0-cuda11.8-cudnn9-runtime
+      docker.io/tensorflow/tensorflow:2.19.0-gpu
+      docker.io/tensorflow/tensorflow:2.18.0-gpu
+      docker.io/tensorflow/tensorflow:2.17.0-gpu
 
   registry.cern.ch/kubeflow/charts:
     type: chart
-- 
GitLab