Compare commits

...

2 Commits

Author SHA1 Message Date
ysh
8084dbabd4 Merge pull request 'GPU正式可用版本' (#19) from dev1 into main
Reviewed-on: #19
2025-07-18 21:32:16 +08:00
ysh
9532036e54 GPU正式可用版本 2025-07-18 21:31:49 +08:00

View File

@ -302,7 +302,7 @@ def create_or_update_statefulset(apps_v1, v1, namespace_name, statefulset_name,
"template": {
"metadata": {"labels": source_selflabel},
"spec": {
"runtimeClassName": "nvidia", # 指定使用 nvidia runtime"
**({"runtimeClassName": "nvidia"} if source_gpu > 0 else {}), # 指定使用 nvidia runtime"
"serviceAccountName": service_account_name,
"restartPolicy": restart_policy,
"containers": [container_config],