一晚上什么事都没做,就一直在查找tensorflow1.4缺少cudnn64_6的错误。
最后发现自己一直用的是cudnn64_7,即使改成cudnn64_6仍不可行。主要受实验室大佬安装的cudnn64_7的影响,估计它用的是cuda9。最后重新下载安装了cudnn64_6后代码跑成功了。
撒花!!!
附上错误语句
I tensorflow/core/common_runtime/gpu/gpu_device.cc:838] Creating TensorFlow device(/gpu:0) -> (device: 0, name: Tesla K20m, pci bus id: 0000:02:00.0)
E tensorflow/stream_executor/cuda/cuda_dnn.cc:347] Loaded runtime CuDNN library: 6022 (compatibility version 5000) but source was compiled with 5103 (compatibility version 5100). If using a binary install, upgrade your CuDNN library to match. If building from sources, make sure the library loaded at runtime matches a compatible version specified during compile configuration.