diff --git a/comfyui-rocm-dockerfile b/comfyui-rocm-dockerfile index a0e08fe..f3311a8 100644 --- a/comfyui-rocm-dockerfile +++ b/comfyui-rocm-dockerfile @@ -18,9 +18,9 @@ RUN apt-get update && apt-get install -y \ WORKDIR /app RUN git clone --recursive https://github.com/ollama/ollama.git . -# Configure and build using Ollama's native Vulkan backend flag +# Configure and safely build with constrained parallelism to protect 96GB RAM RUN cmake -B build . -DOLLAMA_LLAMA_BACKENDS="vulkan" -DCMAKE_BUILD_TYPE=Release -RUN cmake --build build --parallel $(nproc) +RUN cmake --build build --parallel 16 # Build the main Go application binary RUN go build -o ollama .