Skip to content

Commit 5044ec3

Browse files
authored
Upgrade PyTorch to 1.11 (#1155)
* Upgrade PyTorch to 1.11 http://b/215031404
1 parent 0c6ce51 commit 5044ec3

File tree

3 files changed

+10
-6
lines changed

3 files changed

+10
-6
lines changed

Dockerfile.tmpl

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -106,7 +106,7 @@ RUN conda install -c pytorch magma-cuda${CUDA_MAJOR_VERSION}${CUDA_MINOR_VERSION
106106
rm -rf /tmp/torch && \
107107
/tmp/clean-layer.sh
108108
{{ else }}
109-
RUN pip install torch==$TORCH_VERSION+cpu torchvision==$TORCHVISION_VERSION+cpu torchaudio==$TORCHAUDIO_VERSION torchtext==$TORCHTEXT_VERSION -f https://download.pytorch.org/whl/torch_stable.html && \
109+
RUN pip install torch==$TORCH_VERSION+cpu torchvision==$TORCHVISION_VERSION+cpu torchaudio==$TORCHAUDIO_VERSION+cpu torchtext==$TORCHTEXT_VERSION -f https://download.pytorch.org/whl/torch_stable.html && \
110110
/tmp/clean-layer.sh
111111
{{ end }}
112112

@@ -497,7 +497,8 @@ RUN pip install flashtext && \
497497
pip install bqplot && \
498498
pip install earthengine-api && \
499499
pip install transformers && \
500-
pip install datasets && \
500+
# b/232247930 >= 2.2.0 requires pyarrow >= 6.0.0 which conflicts with dependencies for rapidsai 0.21.*
501+
pip install datasets==2.1.0 && \
501502
pip install dlib && \
502503
pip install kaggle-environments && \
503504
pip install geopandas && \

config.txt

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -3,9 +3,10 @@ BASE_IMAGE_TAG=m91
33
CPU_BASE_IMAGE_NAME=tf2-cpu.2-6
44
GPU_BASE_IMAGE_NAME=tf2-gpu.2-6
55
LIGHTGBM_VERSION=3.3.1
6-
TORCH_VERSION=1.9.1
7-
TORCHAUDIO_VERSION=0.9.1
8-
TORCHTEXT_VERSION=0.10.1
9-
TORCHVISION_VERSION=0.10.1
6+
TORCH_VERSION=1.11.0
7+
# TODO(b/215031404#comment4) Remove zlib sed command after upgrade to >= 0.11.1
8+
TORCHAUDIO_VERSION=0.11.0
9+
TORCHTEXT_VERSION=0.12.0
10+
TORCHVISION_VERSION=0.12.0
1011
CUDA_MAJOR_VERSION=11
1112
CUDA_MINOR_VERSION=0

packages/torch.Dockerfile

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -44,6 +44,8 @@ RUN cd /usr/local/src && \
4444
git checkout tags/v$TORCHAUDIO_VERSION && \
4545
git submodule sync && \
4646
git submodule update --init --recursive --jobs 0 && \
47+
# TODO(b/215031404#comment4) Remove after upgrade next release (0.11.1)
48+
sed -i s?https://zlib.net/zlib-1.2.11.tar.gz?https://sourceforge.net/projects/libpng/files/zlib/1.2.11/zlib-1.2.11.tar.gz? third_party/zlib/CMakeLists.txt && \
4749
python setup.py bdist_wheel
4850

4951
# Build torchtext

0 commit comments

Comments
 (0)