huggingface/text-generation-inference
5
0
Fork 0
mirror of https://github.com/huggingface/text-generation-inference.git synced 2025-06-19 15:52:08 +00:00
Code Issues Packages Projects Releases Wiki Activity

Bump llama.cpp & cuda

Browse Source 
Signed-off-by: Adrien Gallouët <angt@huggingface.co>
This commit is contained in:
Adrien Gallouët 2025-02-06 13:24:36 +00:00
parent 7bff88bba9
commit df723e646b
No known key found for this signature in database
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
Dockerfile_llamacpp
View File

@ -1,6 +1,6 @@
FROM nvidia/cuda:12.6.3-cudnn-devel-ubuntu24.04 AS deps
FROM nvidia/cuda:12.8.0-cudnn-devel-ubuntu24.04 AS deps
ARG llamacpp_version=b4628
ARG llamacpp_version=b4651
ARG llamacpp_cuda=OFF
ARG cuda_arch=75-real;80-real;86-real;89-real;90-real
ENV TGI_LLAMA_PKG_CUDA=cuda-${CUDA_VERSION%.*}
@ -56,7 +56,7 @@ RUN cargo build \
--profile release-opt \
--package text-generation-router-llamacpp --frozen
FROM nvidia/cuda:12.6.3-cudnn-runtime-ubuntu24.04
FROM nvidia/cuda:12.8.0-cudnn-runtime-ubuntu24.04
RUN apt update && apt install -y \
python3-venv \