Update k8s-deployment.yaml

uug-ai · Nov 8, 2024 · a27869d · a27869d
1 parent 7df4e6a
commit a27869d
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/k8s-deployment.yaml b/k8s-deployment.yaml
@@ -20,6 +20,7 @@ spec:
           resources:
             limits:
               nvidia.com/gpu: 1 # requesting a single GPU
+              nvidia.com/gpumem: 512 # Each vGPU contains 512m device memory
           env:
             - name: MODEL_NAME
               value: "yolov8n.pt"
@@ -69,4 +70,4 @@ spec:
             - name: FORWARDING_MEDIA
               value: "True"
             - name: REMOVE_AFTER_PROCESSED
-              value: "True"
+              value: "True"