ImportError: libcudnn when running a TensorFlow program

by

Just download cuDNN 5.1 and follow the steps (Tested on Ubuntu 16.04, CUDA toolkit 8.0 )

$ tar xvzf cudnn-8.0-linux-x64-v5.1-ga.tgz
$ sudo cp -P cuda/include/cudnn.h /usr/local/cuda/include
$ sudo cp -P cuda/lib64/libcudnn* /usr/local/cuda/lib64
$ sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*

Now set Path variables

$ vim ~/.bashrc

export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64"
export CUDA_HOME=/usr/local/cuda

and done

For more details, you can check this site

More Related Contents:

  1. TensorFlow and person recognition in video stream
  2. How to compile Tensorflow with SSE4.2 and AVX instructions?
  3. How to understand static shape and dynamic shape in TensorFlow?
  4. Meaning of buffer_size in Dataset.map , Dataset.prefetch and Dataset.shuffle
  5. Can I use TensorBoard with Google Colab?
  6. How do I disable TensorFlow’s eager execution?
  7. After building TensorFlow from source, seeing libcudart.so and libcudnn errors
  8. Convert a graph proto (pb/pbtxt) to a SavedModel for use in TensorFlow Serving or Cloud ML Engine
  9. How to create dataset in the same format as the FSNS dataset?
  10. Specify connections in NN (in keras)
  11. What are the differences between all these cross-entropy losses in Keras and TensorFlow?
  12. Split a dataset created by Tensorflow dataset API in to Train and Test?
  13. Tensorflow crashes with CUBLAS_STATUS_ALLOC_FAILED
  14. Does model.compile() initialize all the weights and biases in Keras (tensorflow backend)?
  15. Initializing tensorflow Variable with an array larger than 2GB
  16. TensorFlow: How to measure how much GPU memory each tensor takes?
  17. How to convert .pb to TFLite format?
  18. Using Gensim Fasttext model with LSTM nn in keras
  19. How to select rows from a 3-D Tensor in TensorFlow?
  20. What is the proper way to install TensorFlow on Apple M1 in 2022
  21. Simple way to visualize a TensorFlow graph in Jupyter?
  22. ValueError: Shapes (None, 1) and (None, 2) are incompatible
  23. CUDA_ERROR_OUT_OF_MEMORY in tensorflow
  24. Convert between NHWC and NCHW in TensorFlow
  25. How to properly use tf.metrics.accuracy?
  26. How can I convert a trained Tensorflow model to Keras?
  27. How to display custom images in TensorBoard using Keras?
  28. Google Storage (gs) wrapper file input/out for Cloud ML?
  29. parallelising tf.data.Dataset.from_generator
  30. Choosing number of Steps per Epoch

Leave a Comment