Upgrade PyTorch to 1.11

http://b/215031404

Author: rosbo

config.txt

@@ -3,9 +3,10 @@ BASE_IMAGE_TAG=m91
 CPU_BASE_IMAGE_NAME=tf2-cpu.2-6
 GPU_BASE_IMAGE_NAME=tf2-gpu.2-6
 LIGHTGBM_VERSION=3.3.1
-TORCH_VERSION=1.9.1
-TORCHAUDIO_VERSION=0.9.1
-TORCHTEXT_VERSION=0.10.1
-TORCHVISION_VERSION=0.10.1
+TORCH_VERSION=1.11.0
+# TODO(b/215031404#comment4) Remove zlib sed command after upgrade to >= 0.11.1
+TORCHAUDIO_VERSION=0.11.0
+TORCHTEXT_VERSION=0.12.0
+TORCHVISION_VERSION=0.12.0
 CUDA_MAJOR_VERSION=11
 CUDA_MINOR_VERSION=0

packages/torch.Dockerfile

@@ -44,6 +44,8 @@ RUN cd /usr/local/src && \
     git checkout tags/v$TORCHAUDIO_VERSION && \
     git submodule sync && \
     git submodule update --init --recursive --jobs 0 && \
+    # TODO(b/215031404#comment4) Remove after upgrade next release (0.11.1)
+    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 && \
     python setup.py bdist_wheel
 
 # Build torchtext