首页 文章

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

提问于
浏览
-2

我一直试图用GPU支持 Build 张量流大约一个星期,并继续遇到与bazel的安装问题 . 当然,我可以在此过程之外运行我的GPU . Windows 10机器,cuda 8.0,cudnn 5.1

我认为它与我的环境变量有关,而不是指向正确的路径和/或某个文件中引用的路径指向usr / local / bin文件夹而不是C:\ Program Files \ NVIDIA GPU计算工具包\ 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)文件”C:/ users / me / anaconda3 / envs / tensorflow_gpu / tensorflow / third_party / gpus / cuda_configure .bzl“,第803行,在_create_cuda_repository _get_cuda_config(repository_ctx)文件”C:/users/me/anaconda3/envs/tensorflow_gpu/tensorflow/third_party/gpus/cuda_configure.bzl“,第619行,在_get_cuda_config _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安装路径....)文件“C:/ users / me /anaconda3/envs/tensorflow_gpu/tensorflow/third_party/gpus/cuda_configure.bzl“,第125行,在aut中o_configure_fail失败(“%sAuto-配置错误:%s ...)”自动配置错误:找不到cudnn安装路径 . . 信息:经过的时间:21.262秒

2 回答

  • 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中的环境变量 .

    再次感谢您的帮助 .

相关问题