mirror of
https://github.com/huggingface/text-generation-inference.git
synced 2025-04-22 15:32:08 +00:00
update docker
This commit is contained in:
parent
cc0552f8fc
commit
22aaf497b7
@ -65,7 +65,7 @@ RUN apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-ins
|
|||||||
COPY --from=ghcr.io/astral-sh/uv:0.5.31 /uv /uvx /bin/
|
COPY --from=ghcr.io/astral-sh/uv:0.5.31 /uv /uvx /bin/
|
||||||
ENV PATH="$PATH:/root/.local/bin"
|
ENV PATH="$PATH:/root/.local/bin"
|
||||||
RUN uv python install ${PYTHON_VERSION}
|
RUN uv python install ${PYTHON_VERSION}
|
||||||
RUN uv venv --python ${PYTHON_VERSION} && uv pip install torch==${PYTORCH_VERSION} pip setuptools packaging
|
RUN uv venv --python ${PYTHON_VERSION} && uv pip install torch==${PYTORCH_VERSION} torchvision pip setuptools packaging
|
||||||
ENV VIRTUAL_ENV=/usr/src/.venv/
|
ENV VIRTUAL_ENV=/usr/src/.venv/
|
||||||
ENV PATH="$PATH:/usr/src/.venv/bin/"
|
ENV PATH="$PATH:/usr/src/.venv/bin/"
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user