replace LLAMA_CUBLAS with GGML_CUDA

dtrifiro · Oct 15, 2024 · 1997565 · 1997565
1 parent 58c396f
commit 1997565
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/Dockerfile.cuda b/Dockerfile.cuda
@@ -39,7 +39,7 @@ WORKDIR /llama-cpp-python
 COPY pyproject.toml .
 COPY poetry.lock .
 
-ENV CMAKE_ARGS="-DLLAMA_CUBLAS=on"
+ENV CMAKE_ARGS="-DGGML_CUDA=on"
 RUN pip3.11 install poetry && \
     scl enable gcc-toolset-12 -- poetry install