tensorflow-gpu2.0报错:Failed to get convolution algorithm. This is probably because cuDNN…

发现tensorflow2.0-gpu版在运行时可以使用GPU运算全连接层,不可以运行卷积运算Conv2D,报错:

UnknownError: Failed to get convolution algorithm. This is probably because cuDNN failed to initialize, so try looking to see if a warning log message was printed above.
在网上试了很多方法,都不可行,例如动态给gpu分配内存啥的,最后按照这个方法成功解决了,即安装了自己CUDA适应的最新的cuDNN,将cuDNN7.4.5版本换成了7.6.5版本:

链接: https://blog.csdn.net/qq_41185569/article/details/107192347

注:可以在任务管理器的这个地方查看gpu在模型计算上的使用率:

tensorflow-gpu2.0报错:Failed to get convolution algorithm. This is probably because cuDNN...

相关文章:

  • 2022-02-21
  • 2021-05-14
  • 2022-12-23
  • 2022-12-23
  • 2022-12-23
  • 2022-12-23
  • 2022-12-23
  • 2021-07-01
猜你喜欢
  • 2021-12-03
  • 2021-10-01
  • 2021-05-11
  • 2021-12-28
  • 2021-11-17
相关资源
相似解决方案