MVPilgrim commited on
Commit
7fbac2c
·
1 Parent(s): fb8e545
Files changed (1) hide show
  1. Dockerfile +1 -1
Dockerfile CHANGED
@@ -49,7 +49,7 @@ RUN pip3 install --break-system-packages https://files.pythonhosted.org/packages
49
  #RUN FORCE_CMAKE=1 CMAKE_SYSTEM_PROCESSOR=AMD64 pip3 install --break-system-packages --verbose --no-cache-dir llama-cpp-python
50
 
51
  #RUN CMAKE_ARGS="-DLLAMA_CUBLAS=on -DCUDA_PATH=/usr/local/cuda-12.2 -DCUDAToolkit_ROOT=/usr/local/cuda-12.2 -DCUDAToolkit_INCLUDE_DIR=/usr/local/cuda-12/include -DCUDAToolkit_LIBRARY_DIR=/usr/local/cuda-12.2/lib64" FORCE_CMAKE=1 pip install llama-cpp-python - no-cache-dir
52
- RUN CMAKE_ARGS="-DLLAMA_CUBLAS=on FORCE_CMAKE=1 pip install --break-system-packages llama-cpp-python --no-cache-dir
53
 
54
  RUN pip3 install --break-system-packages cffi
55
 
 
49
  #RUN FORCE_CMAKE=1 CMAKE_SYSTEM_PROCESSOR=AMD64 pip3 install --break-system-packages --verbose --no-cache-dir llama-cpp-python
50
 
51
  #RUN CMAKE_ARGS="-DLLAMA_CUBLAS=on -DCUDA_PATH=/usr/local/cuda-12.2 -DCUDAToolkit_ROOT=/usr/local/cuda-12.2 -DCUDAToolkit_INCLUDE_DIR=/usr/local/cuda-12/include -DCUDAToolkit_LIBRARY_DIR=/usr/local/cuda-12.2/lib64" FORCE_CMAKE=1 pip install llama-cpp-python - no-cache-dir
52
+ RUN CMAKE_ARGS="-DLLAMA_CUBLAS=on" FORCE_CMAKE=1 pip install --break-system-packages llama-cpp-python --no-cache-dir
53
 
54
  RUN pip3 install --break-system-packages cffi
55