From 5a81390b24d695e10c9f5fc8df51d5994b1e511f Mon Sep 17 00:00:00 2001
From: Michael Yang <mxyng@pm.me>
Date: Thu, 21 Sep 2023 17:01:07 -0700
Subject: [PATCH] update dockerfile.cuda

---
 Dockerfile.cuda | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/Dockerfile.cuda b/Dockerfile.cuda
index b9993da1..2895220f 100644
--- a/Dockerfile.cuda
+++ b/Dockerfile.cuda
@@ -1,4 +1,6 @@
-FROM nvidia/cuda:12.2.0-devel-ubuntu22.04
+ARG CUDA_VERSION=12.2.0
+
+FROM nvidia/cuda:$CUDA_VERSION-devel-ubuntu22.04
 
 ARG TARGETARCH
 ARG VERSION=0.0.0
@@ -13,9 +15,11 @@ ENV GOARCH=$TARGETARCH
 RUN /usr/local/go/bin/go generate ./... \
     && /usr/local/go/bin/go build -ldflags "-linkmode=external -extldflags='-static' -X=github.com/jmorganca/ollama/version.Version=$VERSION -X=github.com/jmorganca/ollama/server.mode=release" .
 
-FROM nvidia/cuda:12.2.0-runtime-ubuntu22.04
+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