From b2c7186e34ae4effec1c7ce493280b31f6f0eb01 Mon Sep 17 00:00:00 2001
From: Olivier DEBAUCHE <olivier.debauche@uliege.be>
Date: Sat, 27 Jul 2024 02:26:54 +0200
Subject: [PATCH] Update cuda.Dockerfile

---
 Docker/cuda/cuda.Dockerfile | 7 +------
 1 file changed, 1 insertion(+), 6 deletions(-)

diff --git a/Docker/cuda/cuda.Dockerfile b/Docker/cuda/cuda.Dockerfile
index e82f176c..a563842d 100644
--- a/Docker/cuda/cuda.Dockerfile
+++ b/Docker/cuda/cuda.Dockerfile
@@ -33,12 +33,7 @@ ENV CUDA_DOCKER_ARCH=all
 ENV LLAMA_CUBLAS=1
 
 # Install Python dependencies
-RUN python3 -m pip install --upgrade pip \
-    && pip install pytest==8.2.1 cmake==3.29.3 \
-    scikit-build==0.17.6 setuptools==70.0.0 \
-    fastapi==0.111.0 uvicorn==0.30.0 \
-    sse-starlette==2.1.0 pydantic-settings==2.2.1 \
-    starlette-context==0.3.6
+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="-DLLAMA_CUBLAS=on" pip install llama-cpp-python