diff --git a/Dockerfile b/Dockerfile index 46c8197..2bcfb2a 100644 --- a/Dockerfile +++ b/Dockerfile @@ -23,7 +23,7 @@ WORKDIR /build # Clone and build llama.cpp RUN git clone https://github.com/ggerganov/llama.cpp && \ cd llama.cpp && \ - CUDA_DOCKER_ARCH=compute_89 make GGML_CUDA=1 + CUDA_DOCKER_ARCH=compute_86 make GGML_CUDA=1 -j 2 # Set the working directory