Merge branch 'main' into deepspeed-amd-pytorch-version-fix

Use torch 2.5.0 for deepspeed image
2025-10-21 01:23:56 +08:00 · 2025-01-28 11:26:36 +01:00 · 2025-01-28 11:20:28 +01:00
1 changed files with 1 additions and 1 deletions
--- 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'
Author	SHA1	Message	Date
ivarflakstad	ee66bc7683	Merge branch 'main' into deepspeed-amd-pytorch-version-fix	2025-01-28 11:26:36 +01:00
Ivar Flakstad	2b02ba2b0a	Use torch 2.5.0 for deepspeed image	2025-01-28 11:20:28 +01:00