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 . \
|
||||
-DOLLAMA_LLAMA_BACKENDS="vulkan" \
|
||||
-DGGML_VULKAN=ON \
|
||||
-DGGML_CPU_ALL_VARIANTS=OFF \
|
||||
-DCMAKE_BUILD_TYPE=Release
|
||||
RUN cmake --build build --parallel 16
|
||||
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
|
||||
|
||||
ENV OLLAMA_HOST=0.0.0.0:11434
|
||||
|
||||
Reference in new issue
Block a user