Donghyeon Kim donghyeonk
donghyeonk
/ compile_tensorflow_serving.sh
Last active
June 19, 2017 10:30
— forked from jorgemf/compile_tensorflow_serving.sh
Compile TensorFlow Serving with CUDA support (March 2017)
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| #!/bin/bash | |
| INITIAL_PATH=$(pwd) | |
| export TF_NEED_CUDA=1 | |
| export TF_NEED_GCP=1 | |
| export TF_NEED_JEMALLOC=1 | |
| export TF_NEED_HDFS=0 | |
| export TF_NEED_OPENCL=0 | |
| export TF_ENABLE_XLA=0 | |
| export TF_CUDA_VERSION=8.0 |