【问题标题】:Environment variable to use tensorflow without cuDNN doesn't work在没有 cuDNN 的情况下使用 tensorflow 的环境变量不起作用
【发布时间】:2017-06-04 20:51:02
【问题描述】:

我不想在我没有sudo 权限的计算机上安装 cuDNN。对于我的 TensorFlow 模型,我不需要 cuDNN,因此我使用环境变量 TF_USE_CUDNN=0。但是,我得到:

ImportError: libcudnn.so.5: cannot open shared object file: No such file or directory

如何在没有 cuDNN 的情况下使用 TensorFlow?

【问题讨论】:

  • 在不支持 GPU 的情况下安装 TF?
  • 我想使用CUDA但没有cuDNN,根据stackoverflow.com/questions/36854940/…这是可能的
  • 提交错误报告——如果不遵守 TF_USE_CUDNN 则毫无意义。

标签: python-3.x tensorflow cudnn


【解决方案1】:

据我从code that reads the environment variable 及其调用者那里得知,TF_USE_CUDNN 环境变量仅在运行时禁用 cuDNN。由于标准 TensorFlow-on-GPU 发行版的链接方式,它仍然需要您安装 cuDNN 库,因为它将与模块​​的其余部分一起加载。最简单的解决方法是在您的 $HOME/lib 目录中本地安装 cuDNN,并设置 LD_LIBRARY_PATH 以在那里查找库。

【讨论】:

    猜你喜欢
    • 2019-01-10
    • 1970-01-01
    • 1970-01-01
    • 1970-01-01
    • 1970-01-01
    • 1970-01-01
    • 1970-01-01
    • 1970-01-01
    • 1970-01-01
    相关资源
    最近更新 更多