-2

大约一周以来，我一直在尝试使用 GPU 支持构建 tensorflow，并且一直遇到 bazel 的安装问题。当然，我可以在这个过程之外运行我的 GPU。Windows 10 机器，cuda 8.0，cudnn 5.1

我相信这与我的环境变量没有指向正确的路径和/或在指向 usr/local/bin 文件夹而不是 C:\Program Files\ 的文件中引用的路径有关NVIDIA GPU 计算工具包\CUDA\v8.0

如果有人可以提供正确的路径，那将非常有帮助。另一个想法是我是否需要修改 bazel 配置文件来帮助解决这个问题？ 这里的环境路径：

我的错误：

C:\Users\me\Anaconda3\envs\tensorflow_gpu\tensorflow>bazel build -c opt --copt=mavx tensorflow/examples/image_retraining:retrain 。错误：加载包'tensorflow/examples/image_retraining'时出错：读取扩展文件'cuda/build_defs.bzl'时遇到错误：没有这样的包'@local_config_cuda//cuda'：回溯（最后一次调用）：文件“C： /users/me/anaconda3/envs/tensorflow_gpu/tensorflow/third_party/gpus/cuda_configure.bzl”，第 915 行 _create_cuda_repository(repository_ctx) 文件“C:/users/me/anaconda3/envs/tensorflow_gpu/tensorflow/third_party/gpus/cuda_configure .bzl”，第 803 行，在 _create_cuda_repository _get_cuda_config(repository_ctx) 文件“C:

自动配置错误：找不到 cudnn 安装路径。. 信息：经过时间：21.262s

4

2 回答 2

1

我确实解决了这个问题。答案在于windows环境和大概linux环境之间的正斜杠和反斜杠。

我确保在 cuda_configure.bzl 中包含正斜杠

_DEFAULT_CUDA_TOOLKIT_PATH = "C:/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v8.0" _DEFAULT_CUDNN_INSTALL_PATH = "C:/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v8.0"

以及windows中的环境变量。

再次感谢您的帮助。

于 2017-05-17T01:09:45.020 回答
0

尝试设置CUDNN_INSTALL_PATH环境变量： https ://github.com/tensorflow/tensorflow/blob/da0d883cba8c3b8103e17432bf8398a56a719c40/third_party/gpus/cuda_configure.bzl#L16

于 2017-05-16T19:39:18.043 回答