diff --git a/ollama/values.gomplate.yaml b/ollama/values.gomplate.yaml
index 99e0ff8..22462c4 100644
--- a/ollama/values.gomplate.yaml
+++ b/ollama/values.gomplate.yaml
@@ -1,3 +1,7 @@
+{{- if eq .Env.OLLAMA_GPU_ENABLED "true" }}
+runtimeClassName: "nvidia"
+{{- end }}
+
 ollama:
   gpu:
     enabled: {{ if eq .Env.OLLAMA_GPU_ENABLED "true" }}true{{ else }}false{{ end }}