Skip to content

Commit 510c2a0

Browse files
authored
Change scheduled CIs to use torch 1.12.1 (#18644)
Co-authored-by: ydshieh <[email protected]>
1 parent 9cf2746 commit 510c2a0

File tree

3 files changed

+3
-3
lines changed

3 files changed

+3
-3
lines changed

docker/transformers-all-latest-gpu/Dockerfile

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@ SHELL ["sh", "-lc"]
99
# The following `ARG` are mainly used to specify the versions explicitly & directly in this docker file, and not meant
1010
# to be used as arguments for docker build (so far).
1111

12-
ARG PYTORCH='1.12.0'
12+
ARG PYTORCH='1.12.1'
1313
# (not always a valid torch version)
1414
ARG INTEL_TORCH_EXT='1.11.0'
1515
# Example: `cu102`, `cu113`, etc.

docker/transformers-pytorch-deepspeed-latest-gpu/Dockerfile

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@ LABEL maintainer="Hugging Face"
33

44
ARG DEBIAN_FRONTEND=noninteractive
55

6-
ARG PYTORCH='1.12.0'
6+
ARG PYTORCH='1.12.1'
77
# Example: `cu102`, `cu113`, etc.
88
ARG CUDA='cu113'
99

docker/transformers-pytorch-gpu/Dockerfile

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@ RUN git clone https://github.com/huggingface/transformers && cd transformers &&
1212
RUN python3 -m pip install --no-cache-dir -e ./transformers[dev-torch,testing]
1313

1414
# If set to nothing, will install the latest version
15-
ARG PYTORCH='1.12.0'
15+
ARG PYTORCH='1.12.1'
1616
ARG TORCH_VISION=''
1717
ARG TORCH_AUDIO=''
1818

0 commit comments

Comments
 (0)