https://github.com/bobbui/tensorflow-serving-cuda-docker

Docker image for tensorflow serving with Nvidia CUDA, CuDNN
https://github.com/bobbui/tensorflow-serving-cuda-docker

cuda cudnn docker docker-image tensorflow tensorflow-serving ubuntu1604

Last synced: 6 days ago
JSON representation

Docker image for tensorflow serving with Nvidia CUDA, CuDNN

• Host: GitHub
• URL: https://github.com/bobbui/tensorflow-serving-cuda-docker
• Owner: bobbui
• License: apache-2.0
• Created: 2017-02-23T14:28:33.000Z (almost 8 years ago)
• Default Branch: master
• Last Pushed: 2017-09-26T18:14:40.000Z (over 7 years ago)
• Last Synced: 2025-01-18T09:12:07.946Z (13 days ago)
• Topics: cuda, cudnn, docker, docker-image, tensorflow, tensorflow-serving, ubuntu1604
• Language: Shell
• Homepage:
• Size: 12.7 KB
• Stars: 8
• Watchers: 1
• Forks: 0
• Open Issues: 0
• Metadata Files:
  • Readme: README.md
  • License: LICENSE

Awesome Lists containing this project

README

        
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 

# tensorflow-serving-cuda-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: 

- ubuntu 14.04

- CUDA 8 with development packages

- cuDNN 5 with development packages

- bazel 0.4.4

- tensorflow 1.0.0

- tensorflow serving source code that is ready to build 

- other setting: proxy,enviroment variable, etc .

You can using either following dockerfile:

- Dockerfile: no tensorflow serving  build happen:  

- Dockerfile.GUI extended from above Dockerfile with GUI and VNC server:  

    connect to: vnc://:5901 via VNC client. The VNC password is **password**. 

After build the docker image, run tensorflow serving compliation with the desired setting.

```

cd /serving/tensorflow

./configure

cd /serving

bazel build -c opt --config=cuda tensorflow_serving/...

bazel test tensorflow_serving/...

```