Upgrade base image to m64. (#969)

- Pin pyproj since 3.x is not compatible with latest version of basemap
yet.
- Improved the basemap test.

http://b/182405233

Author: rosbo

Dockerfile

@@ -1,4 +1,4 @@
-ARG BASE_TAG=m61
+ARG BASE_TAG=m64
 ARG TENSORFLOW_VERSION=2.4.1
 
 FROM gcr.io/kaggle-images/python-tensorflow-whl:${TENSORFLOW_VERSION}-py37 as tensorflow_whl
@@ -39,8 +39,9 @@ RUN conda config --add channels conda-forge && \
     conda config --add channels pytorch && \
     conda config --add channels rapidsai && \
     # ^ rapidsai is the highest priority channel, default lowest, conda-forge 2nd lowest.
-    # 161473620#comment7 pin required to prevent resolver from picking pysal 1.x., pysal 2.2.x is also downloading data on import.
-    conda install matplotlib basemap cartopy python-igraph imagemagick "pysal==2.1.0" && \
+    # b/182405233 pyproj 3.x is not compatible with basemap 1.2.1
+    # b/161473620#comment7 pin required to prevent resolver from picking pysal 1.x., pysal 2.2.x is also downloading data on import.
+    conda install matplotlib basemap cartopy python-igraph imagemagick "pyproj=2.6" "pysal==2.1.0" && \
     conda install "pytorch=1.7" "torchvision=0.8" "torchaudio=0.7" "torchtext=0.8" cpuonly && \
     /tmp/clean-layer.sh
 

tensorflow-whl/Dockerfile

@@ -1,5 +1,5 @@
 FROM nvidia/cuda:10.2-cudnn7-devel-ubuntu18.04 AS nvidia
-FROM gcr.io/deeplearning-platform-release/base-cpu:m61
+FROM gcr.io/deeplearning-platform-release/base-cpu:m64
 
 # Avoid interactive configuration prompts/dialogs during apt-get.
 ENV DEBIAN_FRONTEND=noninteractive

tests/test_matplotlib.py

@@ -14,5 +14,5 @@ def test_plot(self):
         self.assertTrue(os.path.isfile("plot1.png"))
 
     def test_basemap(self):
-        Basemap(width=100,height=100,projection='aeqd',
-                lat_0=40,lon_0=-105)
+        m = Basemap(width=100,height=100,projection='aeqd', lat_0=40,lon_0=-105)
+        self.assertEqual(0, m.xmin)