当前位置：网站首页>YOLOv5 Unable to find a valid cuDNN algorithm to run convolution

YOLOv5 Unable to find a valid cuDNN algorithm to run convolution

2022-04-23 21:00:00 【Top of the program】

YOLOv5 Encountered bug processing method
When we test yolov5l.pt When this model is pre trained , Will be submitted to the Unable to find a valid cuDNN algorithm to run convolution This mistake , When you reinstall cudnn Still can't solve the problem after . In fact, the problem is very simple , It's not ours cudnn Something is wrong. . It is , because yolov5l Belongs to large model . There may be insufficient video memory , At this time, just turn it down batch_size that will do .

版权声明
本文为[Top of the program]所创，转载请带上原文链接，感谢
https://yzsam.com/2022/111/202204210545091362.html