Docker image for tensorflow serving with Nvidia CUDA, CuDNN
APACHE-2.0 License
OBSOLETE: now either TFS install via package manager like apt-get or using official dockerfile from here https://github.com/tensorflow/serving/tree/master/tensorflow_serving/tools/docker
Docker image for tensorflow serving with CUDA which suitable for development purpose
This image is based from nvidia/cuda:8.0-cudnn5-devel-ubuntu14.04 include:
You can using either following dockerfile:
cd /serving/tensorflow
./configure
cd /serving
bazel build -c opt --config=cuda tensorflow_serving/...
bazel test tensorflow_serving/...