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

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

使用的版本为tensorflow2.0

报错信息：

tensorflow.python.framework.errors_impl.UnknownError:  [_Derived_]  Fail to find the dnn implementation.
	 [[{{node CudnnRNN}}]]
	 [[sequential/lstm/StatefulPartitionedCall]] [Op:__inference_distributed_function_3157]
Function call stack:
distributed_function -> distributed_function -> distributed_function

原因：gpu内存的问题

解决办法：

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

 

相关文章：

https://blog.csdn.net/weixin_44545603/article/details/103322446?utm_medium=distribute.pc_relevant.none-task-blog-OPENSEARCH-1.nonecase&depth_1-utm_source=distribute.pc_relevant.none-task-blog-OPENSEARCH-1.nonecase

来源：oschina

链接：https://my.oschina.net/u/4330611/blog/4443944