GPU正式可用版本 #18

Merged
ysh merged 1 commits from dev1 into main 2025-07-18 21:24:26 +08:00

View File

@ -296,13 +296,13 @@ def create_or_update_statefulset(apps_v1, v1, namespace_name, statefulset_name,
"kind": "StatefulSet", "kind": "StatefulSet",
"metadata": {"name": statefulset_name, "namespace": namespace_name}, "metadata": {"name": statefulset_name, "namespace": namespace_name},
"spec": { "spec": {
"runtimeClassName": "nvidia", # 指定使用 nvidia runtime"
"replicas": replicas, "replicas": replicas,
"selector": {"matchLabels": source_selflabel}, "selector": {"matchLabels": source_selflabel},
"serviceName": statefulset_name, "serviceName": statefulset_name,
"template": { "template": {
"metadata": {"labels": source_selflabel}, "metadata": {"labels": source_selflabel},
"spec": { "spec": {
"runtimeClassName": "nvidia", # 指定使用 nvidia runtime"
"serviceAccountName": service_account_name, "serviceAccountName": service_account_name,
"restartPolicy": restart_policy, "restartPolicy": restart_policy,
"containers": [container_config], "containers": [container_config],