dyld：未加载库：@rpath/libcudart.8.0.dylib，同时在 Mac OSX 上构建 tensorflow

Question

我正在我的 Mac（一个 hackintosh，所以我有一个 GPU，并且已经安装了 CUDA8.0。它可以很好地构建 caffe，所以我确信它可以工作。）我有已经设置了如下环境变量（我把它们放在 .zshrc、.bash_profile 和 .bashrc 中）：

export CUDA_HOME=/usr/local/cuda
export DYLD_LIBRARY_PATH="$DYLD_LIBRARY_PATH:$CUDA_HOME/lib"
export PATH="$CUDA_HOME/bin:$PATH"
export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:$CUDA_HOME/lib:$CUDA_HOME/extras/CUPTI/lib"

./configure 工作正常。然后我使用命令 bazel build -c opt --config=cuda //tensorflow/tools/pip_package:build_pip_package 开始构建。然后我得到这个错误：

    ERROR: /Development/tensorflow/tensorflow/python/BUILD:572:1: Executing genrule //tensorflow/python:array_ops_pygenrule failed: bash failed: error executing command /bin/bash -c ... (remaining 1 argument(s) skipped): com.google.devtools.build.lib.shell.AbnormalTerminationException: Process terminated by signal 5.
dyld: Library not loaded: @rpath/libcudart.8.0.dylib
  Referenced from: /private/var/tmp/_bazel_zarzen/bdf1cb43f3ff02468b610730bd03f348/execroot/tensorflow/bazel-out/host/bin/tensorflow/python/gen_array_ops_py_wrappers_cc
  Reason: image not found
/bin/bash: line 1: 92702 Trace/BPT trap: 5       bazel-out/host/bin/tensorflow/python/gen_array_ops_py_wrappers_cc @tensorflow/python/ops/hidden_ops.txt 1 > bazel-out/local_darwin-opt/genfiles/tensorflow/python/ops/gen_array_ops.py
Target //tensorflow/tools/pip_package:build_pip_package failed to build

我可以确定错过的图书馆在那里。我也尝试安装预构建的二进制文件（我知道它只支持 CUDA7.5，所以我将 PATH 设置为指向 CUDA7.5，但它不起作用。当我尝试 import tensorflow 时，类似错误 Library not loaded: @rpath/libcudart.7.5.dylib，仅更改了版本号）。

我不知道为什么找不到 lib。任何人都可以帮忙吗？或者有什么建议吗？

Answer 1

以下应修复错误。

找到文件 "genrule-setup.sh"。该文件应该在

<tensorflow source dir>/bazel-tensorflow/external/bazel_tools/tools/genrule/

如果此文件的时间戳发生更改，则 bazel 构建将失败并提示文件已损坏。所以在修改这个文件之前记下时间戳

stat genrule-setup.sh

你应该得到这样的输出：

16777220 25929227 -rwxr-xr-x 1 user wheel 0 242 "Oct 12 23:46:28 2016" "Oct 10 21:49:39 2026" "Oct 12 21:49:39 2016" "Oct 12 21:49:38 2016" 4096 8 0 genrule-setup.sh

记下上面输出的第二个时间戳"Oct 10 21:49:39 2026"

编辑 genrule-setup.sh 文件

nano genrule-setup.sh

并在文件末尾添加环境配置

export DYLD_LIBRARY_PATH=/usr/local/cuda/lib

保存并关闭编辑器。

然后把时间戳改成原来的时间戳

touch -t YYYYMMDDhhmm.SS genrule-setup.sh

例如

touch -t 202610102149.39 genrule-setup.sh

最后，创建一个符号link以避免"Segmentation fault: 11"错误

ln -sf /usr/local/cuda/lib/libcuda.dylib /usr/local/cuda/lib/libcuda.1.dylib

现在重新开始构建

bazel build -c opt --config=cuda //tensorflow/tools/pip_package:build_pip_package

Answer 2

当运行 ./configure 时，确保指定正确的cuda 版本。您可以通过运行:

找到它

/usr/local/cuda/bin/nvcc --version

您可以在构建期间使用 action_env 标志将环境变量传递给 bazel。

bazel build -c opt --config=cuda \
    --action_env=DYLD_LIBRARY_PATH=/usr/local/cuda/lib \
    //tensorflow/tools/pip_package:build_pip_package

https://bazel.build/designs/2016/06/21/environment.html

dyld：未加载库：@rpath/libcudart.8.0.dylib，同时在 Mac OSX 上构建 tensorflow

dyld: Library not loaded: @rpath/libcudart.8.0.dylib, while building tensorflow on Mac OSX

macos

tensorflow

building