从源代码编译tensorflow时,找不到libcublas.so.8.0,libcufft.so.8.0和libcurand.so.8.0

时间:2018-01-17 16:58:27

标签: tensorflow

我正在尝试使用GPU支持编译tensorflow。问题是当用bazel构建时,我得到错误:

/usr/bin/ld: warning: libcublas.so.8.0, needed by bazel-out/k8-opt/bin/_solib_local/_U_S_Stensorflow_Scontrib_Slite_Stoco_Ctoco___Utensorflow/libtensorflow_framework.so, not found (try using -rpath or -rpath-link)
/usr/bin/ld: warning: libcufft.so.8.0, needed by bazel-out/k8-opt/bin/_solib_local/_U_S_Stensorflow_Scontrib_Slite_Stoco_Ctoco___Utensorflow/libtensorflow_framework.so, not found (try using -rpath or -rpath-link)
/usr/bin/ld: warning: libcurand.so.8.0, needed by bazel-out/k8-opt/bin/_solib_local/_U_S_Stensorflow_Scontrib_Slite_Stoco_Ctoco___Utensorflow/libtensorflow_framework.so, not found (try using -rpath or -rpath-link)

但是,这些文件确实存在,并且它们位于我告诉配置工具的目录中。它们位于/ usr / local / cuda / lib64中。我能够运行示例,因此它似乎安装正确。我也知道它可以找到该目录中的一些文件,因为如果我给该工具提供了错误的文件路径,它会立即报告找不到文件。

其他类似问题通常通过在〜/ .bashrc中添加行来解决。我想我拥有所有正确的行,但我已将其包括在内。

export CUDA_HOME=/usr/local/cuda-8.0
export LD_LIBRARY=${CUDA_HOME}/lib64:$LD_LIBRARY
export LD_LIBRARY_PATH=${CUDA_HOME}/lib64:$LD_LIBRARY_PATH
export PATH=${CUDA_HOME}/bin:$PATH

任何能够提供有关tensorflow寻找文件或我可能做错的地方的任何见解的人都非常感谢。

0 个答案:

没有答案