在没有 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
以在那里查找库。
我不想在我没有 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
以在那里查找库。