Install the Conda environment:

conda env create -f environment.yml

Then run training:

% python preprocessing.py --data_dir data
% python train.py --data_dir data --model_path export

DOCKER_BUILDKIT=1 docker build -t benjamintanweihao/kubeflow-mnist env -f Dockerfile

docker run -t --rm -p 8501:8501
-v "$PWD/export:/models/mnist"
-e MODEL_NAME=mnist
tensorflow/serving:1.14.0