Update cuda to version 12.5.0

Smartappli · Jul 27, 2024 · e6c6b33 · e6c6b33
1 parent cb09745
commit e6c6b33
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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