From a34e1ad3cf371d402032af5099a056d04d72ea0f Mon Sep 17 00:00:00 2001
From: Daniel Hiltgen <daniel@ollama.com>
Date: Thu, 25 Jan 2024 16:46:01 -0800
Subject: [PATCH] Switch back to ubuntu base

The size increase for rocm support in the standard image is problematic
We'll revisit multiple tags for rocm support in a follow up PR.
---
 Dockerfile | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/Dockerfile b/Dockerfile
index 9767faa3..a58b963a 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -109,7 +109,8 @@ ARG CGO_CFLAGS
 RUN go build .
 
 # Runtime stages
-FROM --platform=linux/amd64 rocm/dev-centos-7:6.0-complete as runtime-amd64
+FROM --platform=linux/amd64 ubuntu:22.04 as runtime-amd64
+RUN apt-get update && apt-get install -y ca-certificates
 COPY --from=build-amd64 /go/src/github.com/jmorganca/ollama/ollama /bin/ollama
 FROM --platform=linux/arm64 ubuntu:22.04 as runtime-arm64
 RUN apt-get update && apt-get install -y ca-certificates