Switch to tensorflow-cpu base image for tf-api

[deliahu]

Description

The tf-api image is based on tensorflow/tensorflow:2.1.0-py3, which includes GPU support. This leads to some warnings, as well as a larger than necessary image size.

Currently the official TensorFlow Dockerhub does not include a cpu-only image for version 2.1.0 (tensorflow/tensorflow#36841).