我正在我的Mac上(一个黑苹果,所以我有一个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
。有人能帮忙吗?或者有什么建议吗?