diff options
author | Asim Shankar <ashankar@google.com> | 2017-02-01 12:47:36 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-02-01 13:10:38 -0800 |
commit | 2f6183f85bf200867d6f44efd7168cf492541627 (patch) | |
tree | ae1b85492a8fcc83ffe0d080b95c6c6eafb73ec4 /tensorflow/tools/lib_package | |
parent | 1151644f5a918d76ef846bdd46d898c8b4f7aa03 (diff) |
C API: Slight tweaks to binary distribution scripts
Helps with #6267
Change: 146277178
Diffstat (limited to 'tensorflow/tools/lib_package')
-rwxr-xr-x | tensorflow/tools/lib_package/libtensorflow_test.sh | 7 |
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 |