我一直在尝试使用GPU支持构建tensorflow大约一周,并继续遇到安装问题

时间:2017-05-16 04:26:13

标签: tensorflow gpu bazel cudnn

我一直在尝试使用GPU支持构建张量流大约一周,并继续遇到与bazel的安装问题。当然,我可以在此过程之外运行我的GPU。 Windows 10机器,cuda 8.0,cudnn 5.1

我认为它与我的环境变量有关,而不是指向正确的路径和/或某个文件中引用的路径指向usr / local / bin文件夹而不是C:\ Program Files \ NVIDIA GPU Computing Toolkit \ CUDA \ v8.0

如果有人能够提供非常有用的正确路径。另一个想法是我是否需要修改bazel配置文件来帮助解决这个问题? Environment paths here:

我的错误:

  

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':Traceback(最近一次调用最后一次):           文件“C:/users/me/anaconda3/envs/tensorflow_gpu/tensorflow/third_party/gpus/cuda_configure.bzl”,第915行                   _create_cuda_repository(repository_ctx)           在_create_cuda_repository中输入文件“C:/users/me/anaconda3/envs/tensorflow_gpu/tensorflow/third_party/gpus/cuda_configure.bzl”,第803行                   _get_cuda_config(repository_ctx)           _get_cuda_config中的文件“C:/users/me/anaconda3/envs/tensorflow_gpu/tensorflow/third_party/gpus/cuda_configure.bzl”,第619行                   _cudnn_install_basedir(repository_ctx)           文件“C:/users/me/anaconda3/envs/tensorflow_gpu/tensorflow/third_party/gpus/cuda_configure.bzl”,第202行,在_cudnn_install_basedir中                   auto_configure_fail(“找不到cudnn安装路径....)           在auto_configure_fail中输入文件“C:/users/me/anaconda3/envs/tensorflow_gpu/tensorflow/third_party/gpus/cuda_configure.bzl”,第125行                   失败(”   %s自动配置错误:%s ...))

     

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

2 个答案:

答案 0 :(得分:0)

答案 1 :(得分:0)

我确实解决了这个问题。答案在于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中的环境变量。

再次感谢您的帮助。