diff --git a/roles/kubernetes-apps/container_engine_accelerator/nvidia_gpu/defaults/main.yml b/roles/kubernetes-apps/container_engine_accelerator/nvidia_gpu/defaults/main.yml index a4f19df49..1b0d7859a 100644 --- a/roles/kubernetes-apps/container_engine_accelerator/nvidia_gpu/defaults/main.yml +++ b/roles/kubernetes-apps/container_engine_accelerator/nvidia_gpu/defaults/main.yml @@ -11,3 +11,4 @@ nvidia_driver_install_ubuntu_container: gcr.io/google-containers/ubuntu-nvidia-d nvidia_driver_install_supported: false nvidia_gpu_device_plugin_container: "k8s.gcr.io/nvidia-gpu-device-plugin@sha256:0842734032018be107fa2490c98156992911e3e1f2a21e059ff0105b07dd8e9e" nvidia_gpu_nodes: [] +nvidia_gpu_device_plugin_memory: 30Mi diff --git a/roles/kubernetes-apps/container_engine_accelerator/nvidia_gpu/templates/k8s-device-plugin-nvidia-daemonset.yml.j2 b/roles/kubernetes-apps/container_engine_accelerator/nvidia_gpu/templates/k8s-device-plugin-nvidia-daemonset.yml.j2 index f0fcb1ea9..a476389bc 100644 --- a/roles/kubernetes-apps/container_engine_accelerator/nvidia_gpu/templates/k8s-device-plugin-nvidia-daemonset.yml.j2 +++ b/roles/kubernetes-apps/container_engine_accelerator/nvidia_gpu/templates/k8s-device-plugin-nvidia-daemonset.yml.j2 @@ -44,10 +44,10 @@ spec: resources: requests: cpu: 50m - memory: 10Mi + memory: {{ nvidia_gpu_device_plugin_memory }} limits: cpu: 50m - memory: 10Mi + memory: {{ nvidia_gpu_device_plugin_memory }} securityContext: privileged: true volumeMounts: