aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/tools/lib_package
diff options
context:
space:
mode:
authorGravatar Asim Shankar <ashankar@google.com>2017-02-01 12:47:36 -0800
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2017-02-01 13:10:38 -0800
commit2f6183f85bf200867d6f44efd7168cf492541627 (patch)
treeae1b85492a8fcc83ffe0d080b95c6c6eafb73ec4 /tensorflow/tools/lib_package
parent1151644f5a918d76ef846bdd46d898c8b4f7aa03 (diff)
C API: Slight tweaks to binary distribution scripts
Helps with #6267 Change: 146277178
Diffstat (limited to 'tensorflow/tools/lib_package')
-rwxr-xr-xtensorflow/tools/lib_package/libtensorflow_test.sh7
1 files changed, 5 insertions, 2 deletions
diff --git a/tensorflow/tools/lib_package/libtensorflow_test.sh b/tensorflow/tools/lib_package/libtensorflow_test.sh
index 6463ecea70..b887079a08 100755
--- a/tensorflow/tools/lib_package/libtensorflow_test.sh
+++ b/tensorflow/tools/lib_package/libtensorflow_test.sh
@@ -42,7 +42,10 @@ ${TAR} -xzf ${TARFILE} -Ctensorflow
${CC} ${CFILE} -Itensorflow/include -Ltensorflow/lib -ltensorflow -oa.out
# Execute it, with the shared library available.
-# DYLD_LIBRARY_PATH is used on OS X, LD_LIBRARY_PATH on Linux
-export DYLD_LIBRARY_PATH=tensorflow/lib
+# DYLD_LIBRARY_PATH is used on OS X, LD_LIBRARY_PATH on Linux.
+#
+# The tests for GPU require CUDA libraries to be accessible, which
+# are in DYLD_LIBRARY_PATH in the test harness for OS X.
+export DYLD_LIBRARY_PATH=tensorflow/lib:${DYLD_LIBRARY_PATH}
export LD_LIBRARY_PATH=tensorflow/lib
./a.out