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

Question

我正在尝试从TensorFlow机器上的源代码构建RedHat模型服务器。我没有sudo特权，cudnn不在默认目录中。在构建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'

Answer 1

在这个(答案)部分提供解决方案，即使它存在于评论部分，以造福于社区。

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

export CUDNN_INSTALL_PATH=<...>