【拔刀吧TensorFlow】Ubuntu系統下Cudnn报错问题

import tensorflow as tf  
import numpy as np  
      
      
input_data = tf.Variable( np.random.rand(10, 6, 6 ,3), dtype = np.float32 )  
filter_data = tf.Variable( np.random.rand(2, 2, 3, 10), dtype = np.float32)  
y = tf.nn.conv2d(input_data, filter_data, strides = [1, 1, 1, 1], padding = 'SAME')  
output,argmax = tf.nn.max_pool_with_argmax(input = y, ksize = [1, 2, 2, 1], strides = [1, 1, 1, 1], padding = 'SAME')  
      
with tf.Session() as sess:  
	sess.run(tf.global_variables_initializer())
	print(sess.run(argmax))
	print(sess.run(tf.shape(argmax)))

在搭建好系统后，尝试运行了一下上面这段代码，结果出现错误：

I tensorflow/stream_executor/dso_loader.cc:135] successfully opened CUDA library libcublas.so.8.0 locally
I tensorflow/stream_executor/dso_loader.cc:126] Couldn't open CUDA library libcudnn.so.5. LD_LIBRARY_PATH: /usr/local/cuda-8.0/lib64:/usr/local/cuda-8.0/lib:
I tensorflow/stream_executor/cuda/cuda_dnn.cc:3517] Unable to load cuDNN DSO
I tensorflow/stream_executor/dso_loader.cc:135] successfully opened CUDA library libcufft.so.8.0 locally
I tensorflow/stream_executor/dso_loader.cc:135] successfully opened CUDA library libcuda.so.1 locally
I tensorflow/stream_executor/dso_loader.cc:135] successfully opened CUDA library libcurand.so.8.0 locally
···
I tensorflow/core/common_runtime/gpu/gpu_device.cc:885] Found device 0 with properties: 
name: Tesla K40m
major: 3 minor: 5 memoryClockRate (GHz) 0.745
pciBusID 0000:82:00.0
Total memory: 11.17GiB
Free memory: 2.08GiB
I tensorflow/core/common_runtime/gpu/gpu_device.cc:906] DMA: 0 
I tensorflow/core/common_runtime/gpu/gpu_device.cc:916] 0:   Y 
I tensorflow/core/common_runtime/gpu/gpu_device.cc:975] Creating TensorFlow device (/gpu:0) -> (device: 0, name: Tesla K40m, pci bus id: 0000:82:00.0)
F tensorflow/stream_executor/cuda/cuda_dnn.cc:222] Check failed: s.ok() could not find cudnnCreate in cudnn DSO; dlerror: /usr/local/python3/lib/python3.5/site-packages/tensorflow/python/_pywrap_tensorflow.so: undefined symbol: cudnnCreate
Aborted (core dumped)

错误代码cc:222,百度发现是缺少libcudnn.so.5 文件，根据其他博主提供的解决办法，是建立新版本的cudnn中比如libcudnn.so.7与5的软连接，指令：

ln -s libcudnn.so.7 libcudnn.so.5

发现还是不行，后来查找资料发现是因为cudnn版本过高（我安装的7.x版本，记不清了），又下载了cudnn v5.1，解决问题。

记录于此，方便以后配置环境时查阅