【问题标题】:Tensorflow build error : Cannot find cudnn.h under ~Tensorflow 构建错误:在 ~ 下找不到 cudnn.h
【发布时间】:2019-02-22 10:16:25
【问题描述】:

我正在尝试在 Redhat 7.5 ppc64le 上使用 bazel 0.15 构建 tensorflow r1.12。

我遇到以下错误。

[u0017649@sys-97184 tensorflow]$ bazel build --config=opt --config=cuda //tensorflow/tools/pip_package:build_pip_package 
...
ERROR: error loading package 'tensorflow/tools/pip_package': Encountered error while reading extension file 'cuda/build_defs.bzl': no such package 
'@local_config_cuda//cuda': Traceback (most recent call last):
    File 
"/home/u0017649/files/tensorflow/third_party/gpus/cuda_configure.bzl", line 1447
            _create_local_cuda_repository(repository_ctx)
    File 
"/home/u0017649/files/tensorflow/third_party/gpus/cuda_configure.bzl", line 1187, in _create_local_cuda_repository
            _get_cuda_config(repository_ctx)
    File 
"/home/u0017649/files/tensorflow/third_party/gpus/cuda_configure.bzl", line 911, in _get_cuda_config
            _cudnn_version(repository_ctx, cudnn_install_base..., ...)
    File 
"/home/u0017649/files/tensorflow/third_party/gpus/cuda_configure.bzl", line 582, in _cudnn_version
            _find_cudnn_header_dir(repository_ctx, cudnn_install_base...)
    File 
"/home/u0017649/files/tensorflow/third_party/gpus/cuda_configure.bzl", line 869, in _find_cudnn_header_dir
            auto_configure_fail(("Cannot find cudnn.h under %s" ...))
    File 
"/home/u0017649/files/tensorflow/third_party/gpus/cuda_configure.bzl", line 317, in auto_configure_fail
            fail(("\n%sCuda Configuration Error:%...)))

Cuda Configuration Error: Cannot find cudnn.h under /usr/local/cuda-9.2/targets/ppc64le-linux/lib

我在 /usr/local/cuda-9.2/targets/ppc64le-linux/lib 下确实有一个 cudnn.h 的软链接,如下所示。

[u0017649@sys-97184 tensorflow]$ ls -l /usr/local/cuda-9.2/targets/ppc64le-linux/lib/cudnn.h
lrwxrwxrwx. 1 root root 57 Feb 20 10:15 /usr/local/cuda-9.2/targets/ppc64le-linux/lib/cudnn.h -> /usr/local/cuda-9.2/targets/ppc64le-linux/include/cudnn.h

请问有什么cmet吗?

【问题讨论】:

    标签: tensorflow


    【解决方案1】:

    阅读tensorflow/third_party/gpus/cuda_configure.bzl后,我可以通过以下方式解决这个问题。

    $ sudo ln -sf /usr/local/cuda-9.2/targets/ppc64le-linux/include/cudnn.h /usr/include/cudnn.h
    

    【讨论】:

      猜你喜欢
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      • 2021-08-01
      • 2016-04-06
      • 2018-11-06
      • 1970-01-01
      • 1970-01-01
      相关资源
      最近更新 更多