2020-05-17

UnknownError: 2 root error(s) found. (0) Unknown: Failed to get …

遇到了两次这个问题，记录一下

1. Pycharm + tensorflow-gpu + cuda10.0 + cudnn7.6

在pycharm上运行程序报该错，采用了网上大多数的定义config 的方法

config = tf.ConfigProto()
config.gpu_options.allow_growth = True
with tf.Session(config = config) as sess:

问题解决。

2. 下午用 jupyter notebook 训练模型的时候也报了相同的错，用了上午的方法不管用，看了一下cmd上的报错
   ** E tensorflow/stream_executor/cuda/cuda_dnn.cc:329] Could not create cudnn handle: CUDNN_STATUS_ALLOC_FAILED**
   感觉好像不是 显存动态分配 的问题，可能是cuda和cudnn版本的问题，抱着试一试的想法，去官网下载了7.4版本的cudnn，替换到对应路径后，再运行jupyter 的代码，发现问题解决了。

虽然7.6和7.4都是匹配的cuda10.0，但其中可能有什么差别吧，反正换了一个cudnn版本解决了！