在没有 cuDNN 的情况下使用 tensorflow 的环境变量不起作用

Environment variable to use tensorflow without cuDNN doesn't work

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

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

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

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