时间戳

首页 > 解决方案 > 实现SRCNN时获取卷积算法失败

keras - 实现SRCNN时获取卷积算法失败

问题描述

UnknownError: Failed to get convolution algorithm.

这可能是因为 cuDNN 初始化失败，所以我尝试查看上面是否打印了警告日志消息。

[[node conv2d_1/convolution (defined at C:\ProgramData\Anaconda3\lib\site-packages\keras\backend\tensorflow_backend.py:3009) ]] [Op:__inference_keras_scratch_graph_899]

函数调用栈： keras_scratch_graph

标签： keras

解决方案

我能够解决这个错误，问题出在我笔记本电脑上的 GPU 上。我有 GTX 1650，它导致了 tensorflow 的错误。

有两种可能的解决方案

  • 使用 TF CPU 而不是 TF GPU
  • 使用 Pytorch，它似乎对我很好。我能够通过错误并使用相同的 GPU 'GTX 1650'

推荐阅读