使用 Linux Ubuntu 17.10 虚拟机,我正在尝试构建具有 GPU 支持的 tensorflow。我已经安装了 Cuda 9.2、cuDNN 7.1.4 和 NCCL 2.2.13 并使用以下 bazel 命令构建 tensorflow:
bazel build --config=opt //tensorflow/tools/pip_package:build_pip_package
这是使用在以下位置找到的非常有用的教程:
http://www.python36.com/how-to-install-tensorflow-gpu-with-cuda-9-2-for-python-on-ubuntu/2/
但是,在那里的评论中,似乎没有人遇到类似的事情
然后我收到以下错误:
正在加载:正在加载:0 个包已加载错误:跳过“//tensorflow/tools/pip_package:build_pip_package”:错误加载包“tensorflow/tools/pip_package”:读取扩展文件“cuda/build_defs.bzl”时遇到错误:没有这样的包'@local_config_cuda//cuda':回溯(最近一次通话最后):文件“/home/conorlewis/tensorflow/third_party/gpus/cuda_configure.bzl”,第1142行_create_local_cuda_repository(repository_ctx)文件“/home/conorlewis/tensorflow/third_party /gpus/cuda_configure.bzl”,第 1014 行,在 _create_local_cuda_repository _find_libs(repository_ctx, cuda_config) 文件“/home/conorlewis/tensorflow/third_party/gpus/cuda_configure.bzl”,第 655 行,在 _find_libs _find_cuda_lib("cublas", repository_ctx, cpu_value, c..., ...) 文件“/home/conorlewis/tensorflow/third_party/gpus/cuda_configure.bzl”,第 587 行,在 _find_cuda_lib auto_configure_fail((“找不到 cuda 库 %s”%...)) 文件“/home/conorlewis/tensorflow/ third_party/gpus/cuda_configure.bzl”,第 210 行,在 auto_configure_fail 中失败((“\n%sCuda 配置错误:%...)))
Cuda 配置错误:找不到 cuda 库 libcublas.so.9.2 警告:目标模式解析失败。错误:加载包'tensorflow/tools/pip_package'时出错:读取扩展文件'cuda/build_defs.bzl'时遇到错误:没有这样的包'@local_config_cuda//cuda':回溯(最近一次调用):文件“/home /conorlewis/tensorflow/third_party/gpus/cuda_configure.bzl”,第 1142 行"/home/conorlewis/tensorflow/third_party/gpus/cuda_configure.bzl",第 655 行,在 _find_libs _find_cuda_lib("cublas", repository_ctx, cpu_value, c..., ...
Cuda 配置错误:找不到 cuda 库 libcublas.so.9.2 信息:经过时间:0.974 秒信息:0 个进程。失败:构建未成功完成(加载了 0 个包) 失败:构建未成功完成(加载了 0 个包)
从我读过的其他论坛上,人们似乎说安装 cuda 9.0 但是,这样做我得到了一个类似的错误,关于无法找到 libcublas.so.9.2
检查我的 cuda 文件夹,我发现 libcublas.so 存在但 libcublas.so.9.2 不存在。我不知道为什么安装时省略了这个文件的正确版本
我能得到的任何帮助将不胜感激。
提前致谢