TensorFlow 模型服务器 GPU 构建 - cudnn 路径答案

【问题标题】：TensorFlow Model Server GPU Build - cudnn pathTensorFlow 模型服务器 GPU 构建 - cudnn 路径
【发布时间】：2020-04-22 16:10:44
【问题描述】：

我正在尝试在 RedHat 机器上从源代码构建 TensorFlow 模型服务器。我没有 sudo 权限，而且 cudnn 不在默认目录中。有没有办法在构建时指定 cudnn 路径？

Cuda Configuration Error: Failed to run find_cuda_config.py: Could not find any cudnn.h matching version '10.0' in any subdirectory:
        ''
        'include'
        'include/cuda'
        'include/*-linux-gnu'
        'extras/CUPTI/include'
        'include/cuda/CUPTI'
of:
        '/lib'
        '/lib64'
        '/opt/beegfs/lib'
        '/usr'
        '/usr/lib64/atlas'
        '/usr/lib64/dyninst'
        '/usr/lib64/mysql'
        '/usr/lib64/tcl8.5'
        '/usr/local/cuda'
        '/usr/local/cuda-10.0/targets/x86_64-linux/lib'
        '/usr/local/cuda-8.0/targets/x86_64-linux/lib'
        '/usr/local/cuda-9.0/targets/x86_64-linux/lib'
        '/usr/local/cuda-9.1/targets/x86_64-linux/lib'

【问题讨论】：

您要构建什么版本？您应该能够在运行 bazel 之前导出 CUDNN_INSTALL_PATH=<...>。
谢谢，它成功了。我正在尝试构建主分支。

标签： tensorflow tensorflow-serving

【解决方案1】：

为了社区的利益，在此（答案）部分提供解决方案，即使它出现在评论部分。

在运行Bazel之前，我们需要使用以下命令指定CUDNN路径，

export CUDNN_INSTALL_PATH=<...>

【讨论】：