diff --git a/Dockerfile b/Dockerfile index a58b963a..7c921df8 100644 --- a/Dockerfile +++ b/Dockerfile @@ -116,11 +116,21 @@ FROM --platform=linux/arm64 ubuntu:22.04 as runtime-arm64 RUN apt-get update && apt-get install -y ca-certificates COPY --from=build-arm64 /go/src/github.com/jmorganca/ollama/ollama /bin/ollama +# Radeon images are much larger so we keep it distinct from the CPU/CUDA image +FROM --platform=linux/amd64 rocm/dev-centos-7:5.7.1-complete as runtime-rocm +RUN update-pciids +COPY --from=build-amd64 /go/src/github.com/jmorganca/ollama/ollama /bin/ollama +EXPOSE 11434 +ENV OLLAMA_HOST 0.0.0.0 + +ENTRYPOINT ["/bin/ollama"] +CMD ["serve"] + FROM runtime-$TARGETARCH EXPOSE 11434 ENV OLLAMA_HOST 0.0.0.0 ENV PATH=/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin -ENV LD_LIBRARY_PATH=/usr/local/nvidia/lib:/usr/local/nvidia/lib64:/opt/rocm/lib: +ENV LD_LIBRARY_PATH=/usr/local/nvidia/lib:/usr/local/nvidia/lib64 ENV NVIDIA_DRIVER_CAPABILITIES=compute,utility ENTRYPOINT ["/bin/ollama"] diff --git a/scripts/build_docker.sh b/scripts/build_docker.sh index ef02a144..40054ca6 100755 --- a/scripts/build_docker.sh +++ b/scripts/build_docker.sh @@ -13,3 +13,13 @@ docker build \ -f Dockerfile \ -t ollama/ollama:$VERSION \ . + +docker build \ + --load \ + --platform=linux/amd64 \ + --build-arg=VERSION \ + --build-arg=GOFLAGS \ + --target runtime-rocm \ + -f Dockerfile \ + -t ollama/ollama:$VERSION-rocm \ + .