From 2ded8ab206d53f13b9353428a5e8548bdeb440fb Mon Sep 17 00:00:00 2001
From: Jeffrey Morgan <jmorganca@gmail.com>
Date: Mon, 25 Sep 2023 23:52:05 -0700
Subject: [PATCH] use `11.8.0` nvidia dockerfile base image for now

---
 Dockerfile | 11 +++--------
 1 file changed, 3 insertions(+), 8 deletions(-)

diff --git a/Dockerfile b/Dockerfile
index 2895220f..df1b18de 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,6 +1,4 @@
-ARG CUDA_VERSION=12.2.0
-
-FROM nvidia/cuda:$CUDA_VERSION-devel-ubuntu22.04
+FROM nvidia/cuda:11.8.0-devel-ubuntu22.04
 
 ARG TARGETARCH
 ARG VERSION=0.0.0
@@ -19,13 +17,10 @@ FROM ubuntu:22.04
 ENV OLLAMA_HOST 0.0.0.0
 
 RUN apt-get update && apt-get install -y ca-certificates
-
-ARG USER=ollama
-ARG GROUP=ollama
-RUN groupadd $GROUP && useradd -m -g $GROUP $USER
+RUN groupadd ollama && useradd -m -g ollama ollama
 
 COPY --from=0 /go/src/github.com/jmorganca/ollama/ollama /bin/ollama
 
-USER $USER:$GROUP
+USER ollama:ollama
 ENTRYPOINT ["/bin/ollama"]
 CMD ["serve"]