From 0cae633ce1fbb6fa82b03f6b57c5765b611a6830 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=C3=81kos=20Hadnagy?= Date: Tue, 22 Jul 2025 12:11:38 +0200 Subject: [PATCH] Bump AMD container for 2.7.1 PyTorch (#39458) * Bump AMD container for 2.7.1 PyTorch * Forgot to update pinned packages --- docker/transformers-pytorch-amd-gpu/Dockerfile | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/docker/transformers-pytorch-amd-gpu/Dockerfile b/docker/transformers-pytorch-amd-gpu/Dockerfile index eaa9a8c673..a1958d593d 100644 --- a/docker/transformers-pytorch-amd-gpu/Dockerfile +++ b/docker/transformers-pytorch-amd-gpu/Dockerfile @@ -1,10 +1,10 @@ -FROM rocm/pytorch:rocm6.4_ubuntu22.04_py3.10_pytorch_release_2.6.0 +FROM rocm/pytorch:rocm6.4.1_ubuntu24.04_py3.12_pytorch_release_2.7.1 LABEL maintainer="Hugging Face" ARG DEBIAN_FRONTEND=noninteractive -ARG TORCH_VISION='0.21.0' -ARG TORCH_AUDIO='2.6.0' +ARG TORCH_VISION='0.22.0' +ARG TORCH_AUDIO='2.7.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 && \