Update comfyui-rocm-dockerfile
This commit is contained in:
1 parent
4a04a0a991
commit
111d40bb12
1 file changed
+1
-21
+1
-21
@@ -83,31 +83,11 @@ ENV CGO_ENABLED=1
|
|||||||
RUN cmake -B build . \
|
RUN cmake -B build . \
|
||||||
-DOLLAMA_LLAMA_BACKENDS="vulkan" \
|
-DOLLAMA_LLAMA_BACKENDS="vulkan" \
|
||||||
-DGGML_VULKAN=ON \
|
-DGGML_VULKAN=ON \
|
||||||
|
-DGGML_CPU_ALL_VARIANTS=OFF \
|
||||||
-DCMAKE_BUILD_TYPE=Release
|
-DCMAKE_BUILD_TYPE=Release
|
||||||
RUN cmake --build build --parallel 16
|
RUN cmake --build build --parallel 16
|
||||||
RUN go build -o ollama .
|
RUN go build -o ollama .
|
||||||
|
|
||||||
# Stage 3: Minimal Runtime Image
|
|
||||||
FROM ubuntu:24.04
|
|
||||||
|
|
||||||
ENV DEBIAN_FRONTEND=noninteractive
|
|
||||||
|
|
||||||
RUN apt-get update && apt-get install -y \
|
|
||||||
ca-certificates \
|
|
||||||
libvulkan1 \
|
|
||||||
vulkan-tools \
|
|
||||||
libexpat1 \
|
|
||||||
libelf1 \
|
|
||||||
libdrm2 \
|
|
||||||
&& rm -rf /var/lib/apt/lists/*
|
|
||||||
|
|
||||||
# Copy compiled Mesa-git libraries and ICD files from Stage 1
|
|
||||||
COPY --from=builder /opt/mesa /opt/mesa
|
|
||||||
|
|
||||||
# Copy compiled Ollama binary from Stage 2
|
|
||||||
COPY --from=builder /app/ollama /app/ollama
|
|
||||||
COPY --from=builder /app/build /app/build
|
|
||||||
|
|
||||||
EXPOSE 11434
|
EXPOSE 11434
|
||||||
|
|
||||||
ENV OLLAMA_HOST=0.0.0.0:11434
|
ENV OLLAMA_HOST=0.0.0.0:11434
|
||||||
|
|||||||
Reference in new issue
Block a user