This is just copy from https://github.com/tensorflow/tensorflow/blob/master/tensorflow/tools/dockerfiles/dockerfiles/gpu.Dockerfile until we get official tensorflow image with python3.6
docker run --runtime=nvidia -it roboarchive/tensorflow-gpu-py36 /bin/bash