Upgrade base image to m88. (#1122)

* Upgrade base image to m88.

Remove unnecessary pin on statsmodels following tsfresh new release
including this fix: blue-yonder/tsfresh#897

http://b/215027867

* pin setuptools to 59.5.0 in torch build

Author: rosbo

Dockerfile.tmpl

@@ -461,8 +461,6 @@ RUN pip install flashtext && \
     # b/205704651 remove install cmd for matrixprofile after version > 1.1.10 is released.
     pip install git+https://github.com/matrix-profile-foundation/matrixprofile.git@6bea7d4445284dbd9700a097974ef6d4613fbca7 && \
     pip install tsfresh && \
-    # b/204442455 unpin once tsfresh is compatible with latest version of statsmodels.
-    pip install statsmodels==0.12.2 && \
     pip install pykalman && \
     pip install optuna && \
     pip install plotly_express && \

config.txt

@@ -1,5 +1,5 @@
 BASE_IMAGE_REPO=gcr.io/deeplearning-platform-release
-BASE_IMAGE_TAG=m87
+BASE_IMAGE_TAG=m88
 CPU_BASE_IMAGE_NAME=tf2-cpu.2-6
 GPU_BASE_IMAGE_NAME=tf2-gpu.2-6
 LIGHTGBM_VERSION=3.3.1

packages/torch.Dockerfile

@@ -11,7 +11,7 @@ ARG TORCHVISION_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 cmake cffi typing_extensions future six requests dataclasses
+RUN conda install astunparse numpy ninja pyyaml mkl mkl-include setuptools==59.5.0 cmake cffi typing_extensions future six requests dataclasses
 
 # By default, it uses the version from version.txt which includes the `a0` (alpha zero) suffix and part of the git hash.
 # This causes dependency conflicts like these: https://paste.googleplex.com/4786486378496000