Update ollama to use gpu
All checks were successful
continuous-integration/drone/push Build is passing

This commit is contained in:
Tyler Perkins 2024-04-12 21:05:57 -04:00
parent 25f8c5b216
commit 7f3bb2c1f3

View File

@ -13,11 +13,15 @@ spec:
labels: labels:
app: ollama app: ollama
spec: spec:
runtimeClassName: nvidia
containers: containers:
- name: ollama - name: ollama
image: ollama/ollama image: ollama/ollama
ports: ports:
- containerPort: 11434 - containerPort: 11434
resources:
limits:
nvidia.com/gpu: 2
volumeMounts: volumeMounts:
- name: ollama-volume - name: ollama-volume
mountPath: /root/.ollama mountPath: /root/.ollama