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

Merge pull request 'GPU正式可用版本' (#19) from dev1 into main

Browse Source 
Reviewed-on: #19
This commit is contained in:
ysh 2025-07-18 21:32:16 +08:00
commit 8084dbabd4
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],