Optimize make process by specifying proper proc number

dusty-nv · May 20, 2024 · c77acec · c77acec
1 parent 365fb63
commit c77acec
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/packages/llm/ollama/Dockerfile b/packages/llm/ollama/Dockerfile
@@ -35,6 +35,7 @@ RUN CMAKE_VERSION=${CMAKE_VERSION} GOLANG_VERSION=${GOLANG_VERSION} sh /opt/olla
 WORKDIR ollama/llm/generate
 ENV LD_LIBRARY_PATH=/usr/local/nvidia/lib:/usr/local/nvidia/lib64:/usr/local/nvidia/compat:${LD_LIBRARY_PATH} \
  CMAKE_CUDA_ARCHITECTURES=${CMAKE_CUDA_ARCHITECTURES}
+RUN sed 's|-j8|-j$(nproc)|' -i gen_common.sh && cat gen_common.sh
 RUN bash gen_linux.sh
 
 WORKDIR /opt/ollama