From 8ac2d75353f4c3eca90f5aba0ab65e31e3b20b1b Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=C3=81kos=20Hadnagy?= Date: Fri, 20 Jun 2025 14:17:21 +0200 Subject: [PATCH] Pin PyTorch extras for AMD containers (#38941) * Pin additional Torch packages * Remove unused def --------- Co-authored-by: ivarflakstad <69173633+ivarflakstad@users.noreply.github.com> --- docker/transformers-pytorch-amd-gpu/Dockerfile | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/docker/transformers-pytorch-amd-gpu/Dockerfile b/docker/transformers-pytorch-amd-gpu/Dockerfile index 7e51233779b..eaa9a8c6730 100644 --- a/docker/transformers-pytorch-amd-gpu/Dockerfile +++ b/docker/transformers-pytorch-amd-gpu/Dockerfile @@ -3,6 +3,9 @@ LABEL maintainer="Hugging Face" ARG DEBIAN_FRONTEND=noninteractive +ARG TORCH_VISION='0.21.0' +ARG TORCH_AUDIO='2.6.0' + RUN apt update && \ apt install -y --no-install-recommends git libsndfile1-dev tesseract-ocr espeak-ng python3 python3-dev python3-pip python3-dev ffmpeg git-lfs && \ apt clean && \ @@ -20,6 +23,7 @@ WORKDIR / ADD https://api.github.com/repos/huggingface/transformers/git/refs/heads/main version.json RUN git clone https://github.com/huggingface/transformers && cd transformers && git checkout $REF +RUN python3 -m pip install --no-cache-dir torchvision==$TORCH_VISION torchaudio==$TORCH_AUDIO RUN python3 -m pip install --no-cache-dir -e ./transformers[dev-torch,testing,video] RUN python3 -m pip uninstall -y tensorflow flax