saymrwulf/transformers
1
0
Fork 0
mirror of https://github.com/saymrwulf/transformers.git synced 2026-05-14 20:58:08 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Use torch 2.5.0 for deepspeed image

Browse source
This commit is contained in:
Ivar Flakstad 2025-01-28 11:20:28 +01:00
parent 96625d85fd
commit 2b02ba2b0a
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
docker/transformers-pytorch-deepspeed-amd-gpu/Dockerfile
View file

@ -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'