Skip to content

Commit 4206406

Browse files
authored
Update cuda.Dockerfile
1 parent 0054984 commit 4206406

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

Docker/cuda/cuda.Dockerfile

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -32,17 +32,17 @@ ENV CUDA_DOCKER_ARCH=all \
3232
CMAKE_ARGS="-DGGML_CUDA=on"
3333

3434
# Installer les deps Python générales
35-
RUN python3 -m pip install --upgrade --no-cache-dir pip wheel && \
36-
pip install --no-cache-dir \
35+
RUN python3 -m venv /opt/venv && \
36+
/opt/venv/bin/pip install --upgrade --no-cache-dir pip wheel && \
37+
/opt/venv/bin/pip install --no-cache-dir \
3738
pytest scikit-build setuptools \
3839
fastapi uvicorn sse-starlette \
3940
pydantic-settings starlette-context
4041

42+
ENV PATH="/opt/venv/bin:${PATH}"
43+
4144
# Installer llama-cpp-python avec CUDA
42-
# (tu peux fixer une version si besoin, ex: ==0.3.2)
43-
RUN pip install --no-cache-dir \
44-
"llama-cpp-python" \
45-
--verbose
45+
RUN pip install --no-cache-dir "llama-cpp-python" --verbose
4646

4747
EXPOSE 8000
4848

0 commit comments

Comments
 (0)