From e6c6b33720c0a6136319a8152c72f6dd29e927fe Mon Sep 17 00:00:00 2001 From: Olivier DEBAUCHE Date: Sat, 27 Jul 2024 13:11:01 +0200 Subject: [PATCH] Update cuda to version 12.5.0 --- Docker/cuda/cuda.Dockerfile | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/Docker/cuda/cuda.Dockerfile b/Docker/cuda/cuda.Dockerfile index bb657f2f..a9eb0e66 100644 --- a/Docker/cuda/cuda.Dockerfile +++ b/Docker/cuda/cuda.Dockerfile @@ -1,4 +1,4 @@ -ARG CUDA_VERSION="12.4.1" +ARG CUDA_VERSION="12.5.0" ARG OS="ubuntu22.04" ARG CUDA_BUILDER_IMAGE="${CUDA_VERSION}-devel-${OS}" @@ -27,7 +27,7 @@ RUN python3 -m venv venv # RUN python3 -m pip install --upgrade pip pytest cmake scikit-build setuptools fastapi uvicorn sse-starlette pydantic-settings starlette-context # Install llama-cpp-python (build with cuda) -RUN CMAKE_ARGS="-DGGML_CUDA=on" venv/bin/pip install .[server]==0.2.83 +RUN CMAKE_ARGS="-DGGML_CUDA=on" venv/bin/pip install .[server] # RUN make clean FROM nvidia/cuda:${CUDA_RUNTIME_IMAGE} as runtime