Upgrade base image to m91 (#1142)

http://b/227193319

Author: rosbo

Dockerfile.tmpl

@@ -72,8 +72,7 @@ RUN conda config --add channels nvidia && \
     conda config --add channels rapidsai && \
     # Base image channel order: conda-forge (highest priority), defaults.
     # End state: rapidsai (highest priority), nvidia, conda-forge, defaults.
-    # b/216162758 Pin mkl which last version breaks spacy.
-    conda install mkl=2021.4.0 cartopy=0.19 imagemagick=7.1 pyproj==3.1.0 && \
+    conda install mkl cartopy=0.19 imagemagick=7.1 pyproj==3.1.0 && \
     /tmp/clean-layer.sh
 
 {{ if eq .Accelerator "gpu" }}
@@ -155,7 +154,7 @@ RUN pip install pycuda && \
 
 RUN pip install pysal && \
     pip install seaborn python-dateutil dask python-igraph && \
-    pip install pyyaml joblib husl geopy ml_metrics mne pyshp && \
+    pip install pyyaml joblib husl geopy mne pyshp && \
     pip install pandas && \
     pip install flax && \
     # Install h2o from source.

config.txt

@@ -1,5 +1,5 @@
 BASE_IMAGE_REPO=gcr.io/deeplearning-platform-release
-BASE_IMAGE_TAG=m88
+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