【error】RuntimeError: Unable to find a valid cuDNN algorithm to run convolution

问题:

RuntimeError: Unable to find a valid cuDNN algorithm to run convolution

解决方法:

减小batch-size的数值即可;
当我们测试yolov5x.pt这个模型进行预训练的时候,会报Unable to find a valid cuDNN algorithm to run convolution这个错误,当你重新安装cudnn后仍然不能解决问题。其实这个问题很简单,并不是我们的cudnn出问题了。而是,由于yolov5x属于大模型。显存可能不足,这时候只需调小batch_size即可。

 

参考

1. 【问题解决】YOLOv5遇到Unable to find a valid cuDNN algorithm to run convolution

posted on   鹅要长大  阅读(3408)  评论(0编辑  收藏  举报

相关博文:
阅读排行:
· 地球OL攻略 —— 某应届生求职总结
· 周边上新:园子的第一款马克杯温暖上架
· Open-Sora 2.0 重磅开源!
· 提示词工程——AI应用必不可少的技术
· .NET周刊【3月第1期 2025-03-02】

导航

< 2025年3月 >
23 24 25 26 27 28 1
2 3 4 5 6 7 8
9 10 11 12 13 14 15
16 17 18 19 20 21 22
23 24 25 26 27 28 29
30 31 1 2 3 4 5

统计

点击右上角即可分享
微信分享提示