python-3.x - 在没有 cuDNN 的情况下使用 tensorflow 的环境变量不起作用

Question

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

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

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

Answer 1

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