我一直在尝试使用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
答案 0 :(得分:0)
尝试设置CUDNN_INSTALL_PATH
环境变量:
https://github.com/tensorflow/tensorflow/blob/da0d883cba8c3b8103e17432bf8398a56a719c40/third_party/gpus/cuda_configure.bzl#L16
答案 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中的环境变量。
再次感谢您的帮助。