使用张量流而不使用cuDNN的环境变量不起作用
问题描述:
我不想在没有sudo
权限的计算机上安装cuDNN。对于我的TensorFlow模型,我不需要cuDNN,因此我使用环境变量TF_USE_CUDNN=0
。但是,我得到:使用张量流而不使用cuDNN的环境变量不起作用
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
以在那里查找库。
通过安装没有GPU支持的TF? –
我想使用CUDA但没有cuDNN,这是可能的根据https://*.com/questions/36854940/tensorflow-gpu-is-cudnn-optional-couldnt-open-cuda-library-libcudnn-so – user3734670
提交一个错误报告 - 如果没有遵守TF_USE_CUDNN,没有意义。 – MaxB