Skip to content

Instantly share code, notes, and snippets.

@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)
#!/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