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

作者：菜鸟追梦旅行 | 2024-06-14 19:52:11

踩

no valid convolution algorithm

正常安装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/菜鸟追梦旅行/article/detail/719400