feat: enable limiting and reserving gpu resources
This commit is contained in:
@@ -32,6 +32,7 @@ func TriggerDockerArgsProcessDeploy(appName string, processType string) error {
|
||||
|
||||
validLimits := map[string]bool{
|
||||
"cpu": true,
|
||||
"nvidia-gpu": true,
|
||||
"memory": true,
|
||||
"memory-swap": true,
|
||||
}
|
||||
@@ -54,6 +55,10 @@ func TriggerDockerArgsProcessDeploy(appName string, processType string) error {
|
||||
parts[1] = "cpus"
|
||||
}
|
||||
|
||||
if parts[1] == "nvidia-gpu" {
|
||||
parts[1] = "gpus"
|
||||
}
|
||||
|
||||
limits[parts[1]] = value
|
||||
}
|
||||
if parts[0] == "reserve" {
|
||||
|
||||
Reference in New Issue
Block a user