【深度学习】RuntimeError: Unable to find a valid cuDNN algorithm to run convolution

作者：很楠不爱3 | 2024-02-27 03:01:55

踩

runtimeerror: unable to find a valid cudnn algorithm to run convolution

正常安装https://blog.csdn.net/x1131230123/article/details/121798776后遇到这个问题咋办？

原因可能是你的显卡显存容不下你的设置，把batch-size改小即可：

python train.py --batch-size 2 --data kevin_firesmoke.yaml --img 1280 --epochs 500 --weight /home/kevin_xie/yolov5_version6/weights/yolov5l6.pt --optimizer "Adam" --save-period 20
1

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/很楠不爱3/article/detail/149509