TensorFlow的C ++版本

Question

我需要使用TensorFlow的C ++ API。因此，似乎有必要从源代码构建TensorFlow，因此我根据https://www.tensorflow.org/install/install_sources执行此操作。我希望拥有支持GPU的版本。我使用大多数默认选项运行./configure。

但是在运行时

  

bazel build --config=opt --config=cuda //tensorflow/tools/pip_package:build_pip_package

发生以下错误：

  

Cuda配置错误：检测到cuDNN版本   /usr/lib/x86_64-linux-gnu/include/cudnn.h(7.0.3）不匹配   TF_CUDNN_VERSION（6）

我以前安装了CuDNN 7，但随后安装了cudnn 6用于tensorflow。我该怎么办？

Answer 1

您可以尝试从环境变量中禁用CuDNN 7，并尝试将cudnn6文件夹中的文件放入CUDA文件夹中的相关文件夹。

这对我有用，希望有所帮助。