Pytorch: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED

RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED

服务器配置一览: RTX 2080ti,cuda9

所遇问题

在使用pytorch中预训练好的inception_v3网络跑自己实验的时候出现

解决办法:

网上常见的方法汇总:

1. 重装cuda版本 (RTX20180ti 与cuda9的适用性不如cuda10)
2. torch.backends.cudnn.enabled = False (默认为ture，在调用包的后面加入即可)

我使用的方法

比较懒，所以方法2

注意

(摘录) 关于torch.backends.cudnn.enabled 的使用：

设置这个 flag 可以让内置的 cuDNN 的 auto-tuner 自动寻找最适合当前配置的高效算法，来达到优化运行效率的问题。

应该遵循以下准则：

如果网络的输入数据维度或类型上变化不大，置为true 可以增加运行效率；
如果网络的输入数据在每次 iteration 都变化的话，会导致 cnDNN 每次都会去寻找一遍最优配置，这样反而会降低运行效率。

---

转载请注明来源，欢迎对文章中的引用来源进行考证，欢迎指出任何有错误或不够清晰的表达。可以在下面评论区评论，也可以邮件至 luckiestman0@qq.com