mirror/cube-studio
2
0
Fork 0
mirror of https://github.com/tencentmusic/cube-studio.git synced 2024-12-15 06:09:57 +08:00
Code Issues Packages Projects Releases Wiki Activity

添加gpu资源采集

Browse Source
This commit is contained in:
data-infra 2024-01-05 11:27:45 +08:00
parent 4a123245b3
commit da7e55f949
2 changed files with 7 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
install/docker/config.py
View File

@ -680,8 +680,9 @@ GLOBAL_ENV={
"KFJ_ENVIRONMENT":"{{cluster_name}}",
}
# 各种环节使用的gpu驱动类型
GPU_RESOURCE={}
GPU_RESOURCE={
"gpu":"nvidia.com/gpu"
}
DEFAULT_GPU_RESOURCE_NAME='nvidia.com/gpu'
# 配置禁用gpu的方法，不然对复合共用型机器，gpu会被共享使用

6
install/kubernetes/cube/overlays/config/config.py
View File

@ -680,8 +680,10 @@ GLOBAL_ENV={
"KFJ_ENVIRONMENT":"{{cluster_name}}",
}
# 各种环节使用的gpu驱动类型
GPU_RESOURCE={}
GPU_RESOURCE={
"gpu":"nvidia.com/gpu"
}
DEFAULT_GPU_RESOURCE_NAME='nvidia.com/gpu'
# 配置禁用gpu的方法，不然对复合共用型机器，gpu会被共享使用