Upgrade PyTorch to 1.12

http://b/238238619

Author: rosbo

config.txt

@@ -3,10 +3,9 @@ BASE_IMAGE_TAG=m94
 CPU_BASE_IMAGE_NAME=tf2-cpu.2-9
 GPU_BASE_IMAGE_NAME=tf2-gpu.2-9
 LIGHTGBM_VERSION=3.3.2
-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
+TORCH_VERSION=1.12.0
+TORCHAUDIO_VERSION=0.12.0
+TORCHTEXT_VERSION=0.13.0
+TORCHVISION_VERSION=0.13.0
 CUDA_MAJOR_VERSION=11
 CUDA_MINOR_VERSION=0

packages/torch.Dockerfile

@@ -13,7 +13,7 @@ ARG CUDA_MINOR_VERSION
 RUN test -n "$TORCHVISION_VERSION"
 
 # Build instructions: https://github.com/pytorch/pytorch#from-source
-RUN conda install astunparse numpy ninja pyyaml mkl mkl-include setuptools==59.5.0 cmake cffi typing_extensions future six requests dataclasses
+RUN conda install astunparse numpy ninja pyyaml mkl mkl-include setuptools cmake cffi typing_extensions future six requests dataclasses
 RUN conda install -c pytorch magma-cuda${CUDA_MAJOR_VERSION}${CUDA_MINOR_VERSION}
 
 # By default, it uses the version from version.txt which includes the `a0` (alpha zero) suffix and part of the git hash.
@@ -44,8 +44,6 @@ 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