From 4925f2c4763702902562ac0c20d8855c0b185a27 Mon Sep 17 00:00:00 2001
From: ysh <y13141005621@163.com>
Date: Fri, 18 Jul 2025 21:24:05 +0800
Subject: [PATCH] =?UTF-8?q?GPU=E6=AD=A3=E5=BC=8F=E5=8F=AF=E7=94=A8?=
 =?UTF-8?q?=E7=89=88=E6=9C=AC?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 app/k8sManager/k8s_utils_linuxos_ubuntu.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/app/k8sManager/k8s_utils_linuxos_ubuntu.py b/app/k8sManager/k8s_utils_linuxos_ubuntu.py
index 413246f..0dd00b8 100644
--- a/app/k8sManager/k8s_utils_linuxos_ubuntu.py
+++ b/app/k8sManager/k8s_utils_linuxos_ubuntu.py
@@ -296,13 +296,13 @@ def create_or_update_statefulset(apps_v1, v1, namespace_name, statefulset_name,
                 "kind": "StatefulSet",
                 "metadata": {"name": statefulset_name, "namespace": namespace_name},
                 "spec": {
-                    "runtimeClassName": "nvidia",  # 指定使用 nvidia runtime"
                     "replicas": replicas,
                     "selector": {"matchLabels": source_selflabel},
                     "serviceName": statefulset_name,
                     "template": {
                         "metadata": {"labels": source_selflabel},
                         "spec": {
+                            "runtimeClassName": "nvidia",  # 指定使用 nvidia runtime"
                             "serviceAccountName": service_account_name,
                             "restartPolicy": restart_policy,
                             "containers": [container_config],
-- 
2.34.1