在没有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

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