yumoqing/pcapi
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

GPU正式可用版本

Browse Source
This commit is contained in:
ysh 2025-07-18 21:31:49 +08:00
parent 4925f2c476
commit 9532036e54
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
app/k8sManager/k8s_utils_linuxos_ubuntu.py
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],