File tree Expand file tree Collapse file tree 3 files changed +3
-3
lines changed
transformers-all-latest-gpu
transformers-pytorch-deepspeed-latest-gpu Expand file tree Collapse file tree 3 files changed +3
-3
lines changed Original file line number Diff line number Diff 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)
1414ARG INTEL_TORCH_EXT='1.11.0'
1515# Example: `cu102`, `cu113`, etc.
Original file line number Diff line number Diff line change @@ -3,7 +3,7 @@ LABEL maintainer="Hugging Face"
33
44ARG DEBIAN_FRONTEND=noninteractive
55
6- ARG PYTORCH='1.12.0 '
6+ ARG PYTORCH='1.12.1 '
77# Example: `cu102`, `cu113`, etc.
88ARG CUDA='cu113'
99
Original file line number Diff line number Diff line change @@ -12,7 +12,7 @@ RUN git clone https://github.com/huggingface/transformers && cd transformers &&
1212RUN 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 '
1616ARG TORCH_VISION=''
1717ARG TORCH_AUDIO=''
1818
You can’t perform that action at this time.
0 commit comments