瀏覽代碼

Merge pull request #2196 from dhiltgen/remove_rocm_image

Switch back to ubuntu base
Daniel Hiltgen 1 年之前
父節點
當前提交
197e420a97
共有 1 個文件被更改,包括 2 次插入1 次删除
  1. 2 1
      Dockerfile

+ 2 - 1
Dockerfile

@@ -109,7 +109,8 @@ ARG CGO_CFLAGS
 RUN go build .
 RUN go build .
 
 
 # Runtime stages
 # 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
 COPY --from=build-amd64 /go/src/github.com/jmorganca/ollama/ollama /bin/ollama
 FROM --platform=linux/arm64 ubuntu:22.04 as runtime-arm64
 FROM --platform=linux/arm64 ubuntu:22.04 as runtime-arm64
 RUN apt-get update && apt-get install -y ca-certificates
 RUN apt-get update && apt-get install -y ca-certificates