From 2b02ba2b0ae90b4800e1731b446845221c0fc442 Mon Sep 17 00:00:00 2001
From: Ivar Flakstad <69173633+ivarflakstad@users.noreply.github.com>
Date: Tue, 28 Jan 2025 11:20:28 +0100
Subject: [PATCH] Use torch 2.5.0 for deepspeed image

---
 docker/transformers-pytorch-deepspeed-amd-gpu/Dockerfile | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/docker/transformers-pytorch-deepspeed-amd-gpu/Dockerfile b/docker/transformers-pytorch-deepspeed-amd-gpu/Dockerfile
index a8edb8ff0..67db9823f 100644
--- a/docker/transformers-pytorch-deepspeed-amd-gpu/Dockerfile
+++ b/docker/transformers-pytorch-deepspeed-amd-gpu/Dockerfile
@@ -2,7 +2,7 @@ FROM rocm/dev-ubuntu-22.04:6.2.4
 LABEL maintainer="Hugging Face"
 
 ARG DEBIAN_FRONTEND=noninteractive
-ARG PYTORCH='2.5.1'
+ARG PYTORCH='2.5.0'
 ARG TORCH_VISION='0.20.0'
 ARG TORCH_AUDIO='2.5.0'
 ARG ROCM='6.2'