diff --git a/ollama/ollama-deployment.yaml b/ollama/ollama-deployment.yaml
index 409d010..dea9cf3 100644
--- a/ollama/ollama-deployment.yaml
+++ b/ollama/ollama-deployment.yaml
@@ -13,11 +13,15 @@ spec:
       labels:
         app: ollama
     spec:
+      runtimeClassName: nvidia
       containers:
       - name: ollama
         image: ollama/ollama
         ports:
         - containerPort: 11434
+        resources:
+          limits:
+            nvidia.com/gpu: 2
         volumeMounts:
         - name: ollama-volume
           mountPath: /root/.ollama