2017-06-04 77 views
-1

我不想在没有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?

+0

通过安装没有GPU支持的TF? –

+0

我想使用CUDA但没有cuDNN,这是可能的根据https://stackoverflow.com/questions/36854940/tensorflow-gpu-is-cudnn-optional-couldnt-open-cuda-library-libcudnn-so – user3734670

+0

提交一个错误报告 - 如果没有遵守TF_USE_CUDNN,没有意义。 – MaxB

回答

2

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