fix docker build script

ollama · Apr 9, 2024 · 6b1fb68 · 6b1fb68
1 parent 65ae7c0
commit 6b1fb68
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -107,7 +107,7 @@ COPY --from=cpu_avx2-build-amd64 /go/src/github.com/ollama/ollama/llm/build/linu
 COPY --from=cuda-build-amd64 /go/src/github.com/ollama/ollama/llm/build/linux/ llm/build/linux/
 COPY --from=rocm-build-amd64 /go/src/github.com/ollama/ollama/llm/build/linux/ llm/build/linux/
 COPY --from=rocm-build-amd64 /go/src/github.com/ollama/ollama/dist/deps/ ./dist/deps/
-COPY --from=oneapi-build-amd64 /go/src/github.com/ollama/ollama/llm/llama.cpp/build/linux/ llm/llama.cpp/build/linux/
+COPY --from=oneapi-build-amd64 /go/src/github.com/ollama/ollama/llm/build/linux/ llm/build/linux/
 ARG GOFLAGS
 ARG CGO_CFLAGS
 RUN go build -trimpath .
@@ -143,7 +143,8 @@ ENTRYPOINT ["/bin/ollama"]
 CMD ["serve"]
 
 # oneAPI images are much larger so we keep it distinct from the CPU/CUDA image
-FROM --platform=linux/amd64 intel/oneapi-runtime:2024.0.1-devel-rockylinux9 as runtime-oneapi
+FROM --platform=linux/amd64 intel/oneapi-basekit:2024.0.1-devel-rockylinux9 as runtime-oneapi
+RUN update-pciids
 COPY --from=build-amd64 /go/src/github.com/ollama/ollama/ollama /bin/ollama
 EXPOSE 11434
 ENV OLLAMA_HOST 0.0.0.0

diff --git a/scripts/build_docker.sh b/scripts/build_docker.sh
@@ -85,5 +85,6 @@ if [ -z "${OLLAMA_SKIP_MANIFEST_CREATE}" ]; then
         echo "  ${RELEASE_IMAGE_REPO}:$VERSION-amd64"
         echo "  ${RELEASE_IMAGE_REPO}:$VERSION-arm64"
         echo "  ${RELEASE_IMAGE_REPO}:$VERSION-rocm"
+        echo "  ${RELEASE_IMAGE_REPO}:$VERSION-oneapi"
     fi
 fi