Search code examples
python-3.xtensorflowcudnn

Environment variable to use tensorflow without cuDNN doesn't work


I don't want to install cuDNN on a computer for which I haven't sudo privileges. For my TensorFlow model I don't need cuDNN, therefore I use the environment variable TF_USE_CUDNN=0. However, I get:

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

How I can use TensorFlow without cuDNN?


Solution

  • As far as I can tell from the code that reads the environment variable and its callers, the TF_USE_CUDNN environment variable only disables the use of cuDNN at runtime. Due to the way the standard TensorFlow-on-GPU distribution is linked, it still requires you to install the cuDNN library, because it will be loaded with the rest of the module. The easiest workaround would be to install cuDNN locally in your $HOME/lib directory, and set up LD_LIBRARY_PATH to look for the library there.