diff --git a/huggingface/pytorch/tgi/docker/2.0.3/Dockerfile b/huggingface/pytorch/tgi/docker/2.0.3/Dockerfile index deb5ecc..e5a00ba 100644 --- a/huggingface/pytorch/tgi/docker/2.0.3/Dockerfile +++ b/huggingface/pytorch/tgi/docker/2.0.3/Dockerfile @@ -51,7 +51,7 @@ ARG TARGETPLATFORM ENV PATH /opt/conda/bin:$PATH -RUN apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends \ +RUN apt-get update && apt-get upgrade -y && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends \ build-essential \ ca-certificates \ ccache \ @@ -176,7 +176,7 @@ ENV HUGGINGFACE_HUB_CACHE=/tmp \ WORKDIR /usr/src -RUN apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends \ +RUN apt-get update && apt-get upgrade -y && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends \ libssl-dev \ ca-certificates \ make \ diff --git a/releases.json b/releases.json index 466a1d4..0448644 100644 --- a/releases.json +++ b/releases.json @@ -109,11 +109,12 @@ { "framework": "TGI", "device": "gpu", - "version": "3.1.1", + "version": "2.0.3", "os_version": "ubuntu22.04", - "cuda_version": "cu124", - "python_version": "py311", - "pytorch_version": "2.6.0" + "cuda_version": "cu121", + "python_version": "py310", + "pytorch_version": "2.3.0" + } ] }