Docker images already support CPU based tensorflow processing, but do not support GPU processing.

potentially useful: https://hub.docker.com/r/nvidia/cuda/

how tensorflow does it themselves: https://github.com/tensorflow/tensorflow/blob/master/tensorflow/tools/docker/Dockerfile.devel-gpu