muryshev commited on
Commit
e4894f0
·
1 Parent(s): 599f23b

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -2
Dockerfile CHANGED
@@ -17,11 +17,11 @@ RUN git clone https://github.com/ggerganov/llama.cpp.git
17
  WORKDIR /build/llama.cpp
18
 
19
  ENV CUDA_DOCKER_ARCH=${CUDA_DOCKER_ARCH}
20
- #ENV LLAMA_CUBLAS=1
21
  # cmake .. -DLLAMA_CUBLAS=ON && \
22
  RUN mkdir build && \
23
  cd build && \
24
- cmake .. && \
25
  cmake --build . --config Release
26
 
27
  WORKDIR /data
 
17
  WORKDIR /build/llama.cpp
18
 
19
  ENV CUDA_DOCKER_ARCH=${CUDA_DOCKER_ARCH}
20
+ ENV LLAMA_CUBLAS=1
21
  # cmake .. -DLLAMA_CUBLAS=ON && \
22
  RUN mkdir build && \
23
  cd build && \
24
+ cmake .. -DLLAMA_CUBLAS=ON && \
25
  cmake --build . --config Release
26
 
27
  WORKDIR /data