qq_42174942的博客

原创 could not create cudnn handle: CUDNN_STATUS_INTERNAL_ERROR

在跑深度学习代码时，我们可能会遇到以下一种情况，源错误代码如下： 以上问题在GPU上运行代码，可能是显卡不够用，但在CPU上运行是正常的，我们知道CPU和GPU运行代码的速度相差是非常大的，这时我们只需在源代码上添加两行代码就可以解决这个问题，意思是在一开始占用显卡的内存不要太高，然后按需求慢慢增加内存。添加的代码如下：config = tf.ConfigProto(g...