Merge pull request abetlen#536 from Isydmr/patch-2

abetlen · web-flow · commit d89bf8c0722e · 2023-07-28T01:37:23.000-04:00
Fix OpenBLAS Docker build
diff --git a/docker/openblas_simple/Dockerfile b/docker/openblas_simple/Dockerfile
@@ -9,7 +9,7 @@ COPY . .
 RUN apt update && apt install -y libopenblas-dev ninja-build build-essential
 RUN python -m pip install --upgrade pip pytest cmake scikit-build setuptools fastapi uvicorn sse-starlette pydantic-settings
 
-RUN LLAMA_OPENBLAS=1 pip install llama_cpp_python --verbose
+RUN CMAKE_ARGS="-DLLAMA_BLAS=ON -DLLAMA_BLAS_VENDOR=OpenBLAS" pip install llama_cpp_python --verbose
 
 # Run the server
 CMD python3 -m llama_cpp.server
