Lens icon indicating copy to clipboard operation
Lens copied to clipboard

GPU Enable Docker Images

Open bonedaddy opened this issue 6 years ago • 0 comments

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

bonedaddy avatar Jun 19 '19 03:06 bonedaddy