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