Fail to find the dnn implementation. [Op:CudnnRNN]解决办法

报这个错误基本上就可以确定是gpu中的显存不够了,在这里我发现之前有一个占用gpu资源特别大的程序没有关闭,我将那个对应的程序关闭之后再次运行当前的这个程序,错误立刻就消失了。
网上的策略是实现gpu的显存自增长

# 设置gpu内存自增长
gpus = tf.config.experimental.list_physical_devices('GPU')
print(gpus)
for gpu in gpus:
    tf.config.experimental.set_memory_growth(gpu, True)

能看出来这个问题本质上就是显存带来的问题,只要能够合理分配显存就能够及时地解决问题

已标记关键词 清除标记
相关推荐
©️2020 CSDN 皮肤主题: 书香水墨 设计师:CSDN官方博客 返回首页