From ec7790f0d3034b7a2a03b9d7f00154db1e17e84c Mon Sep 17 00:00:00 2001 From: Yih-Dar <2521628+ydshieh@users.noreply.github.com> Date: Wed, 29 Jan 2025 16:01:27 +0100 Subject: [PATCH] update docker file `transformers-pytorch-deepspeed-latest-gpu` (#35940) update docker file for deepspeed Co-authored-by: ydshieh --- docker/transformers-pytorch-deepspeed-latest-gpu/Dockerfile | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docker/transformers-pytorch-deepspeed-latest-gpu/Dockerfile b/docker/transformers-pytorch-deepspeed-latest-gpu/Dockerfile index eeaf728cab7..36f8506a476 100644 --- a/docker/transformers-pytorch-deepspeed-latest-gpu/Dockerfile +++ b/docker/transformers-pytorch-deepspeed-latest-gpu/Dockerfile @@ -1,5 +1,5 @@ # https://docs.nvidia.com/deeplearning/frameworks/pytorch-release-notes/rel-23-11.html#rel-23-11 -FROM nvcr.io/nvidia/pytorch:23.04-py3 +FROM nvcr.io/nvidia/pytorch:23.11-py3 LABEL maintainer="Hugging Face" ARG DEBIAN_FRONTEND=noninteractive