Build new TF whl

Author: rosbo

tensorflow-whl/CHANGELOG.md

@@ -19,4 +19,5 @@
 * `2.3.0-py37`: TensorFlow 2.3.0 with Python 3.7
 * `2.3.1-py37`: TensorFlow 2.3.1 with Python 3.7
 * `2.3.1-py37-2`: TensorFlow 2.3.1 & TFA 0.11 with Python 3.7
-* `2.4.0-py37`: TensorFlow 2.4.0 & Python 3.7 & TFA 0.12
+* `2.4.0-py37`: TensorFlow 2.4.0 & Python 3.7 & TFA 0.12
+* `2.4.0-py37-2`: TensorFlow 2.4.0 & m61 base image

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:m46
+FROM gcr.io/deeplearning-platform-release/base-cpu:m61
 
 # Avoid interactive configuration prompts/dialogs during apt-get.
 ENV DEBIAN_FRONTEND=noninteractive