diff --git a/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/utils/K8sClientUtil.java b/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/utils/K8sClientUtil.java index 4dd8e6d..0c8fcc4 100644 --- a/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/utils/K8sClientUtil.java +++ b/ruoyi-modules/management-platform/src/main/java/com/ruoyi/platform/utils/K8sClientUtil.java @@ -461,10 +461,11 @@ public class K8sClientUtil { int cpu = (int) valueJson.get("cpu"); String memory = (String) valueJson.get("memory"); memory = memory.substring(0, memory.length() - 1).concat("i"); + int gpu = (int) valueJson.get("gpu"); HashMap limitMap = new HashMap<>(); if (Constant.Computing_Resource_GPU.equals(devEnvironment.getComputingResource())) { - limitMap.put("nvidia.com/gpu", new Quantity("1")); + limitMap.put("nvidia.com/gpu", new Quantity(String.valueOf(gpu))); } limitMap.put("cpu", new Quantity(String.valueOf(cpu))); limitMap.put("memory", new Quantity(memory));