From d49f0c1fb3cf7de6110052f3df4cdaa194539ebd Mon Sep 17 00:00:00 2001
From: ivamp <im@ivampiresp.com>
Date: Wed, 6 Nov 2024 02:03:15 +0800
Subject: [PATCH] =?UTF-8?q?=E6=94=B9=E8=BF=9B?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 manifest.yaml | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

diff --git a/manifest.yaml b/manifest.yaml
index 7784f95..0535df7 100644
--- a/manifest.yaml
+++ b/manifest.yaml
@@ -27,6 +27,7 @@ spec:
         app: text-classification
         tier: backend
     spec:
+      runtimeClassName: "nvidia"
       volumes:
         - name: text-classification-models
           persistentVolumeClaim:
@@ -53,10 +54,10 @@ spec:
             - containerPort: 80
               protocol: TCP
               name: http
-          #          resources:
-          #            requests:
-          #              cpu: 1000m
-          #              memory: 1024Mi
+          resources:
+            limits:
+              nvidia.com/gpu: "1"
+              nvidia.com/gpumem: "1000"
           volumeMounts:
             - mountPath: /app/models
               name: text-classification-models