笔记本运行YOLOv5,3050显卡,python3.8.0 torch-1.12.1 cuda113,能够正常运行检测,每张图像检测速度也很快,但是在代码运行后会“Fusing layers”这一行卡顿很久很久,估计有几分钟的时间,才可以正常运行检测。这里使用的是服务器上训练好的模型,服务器上就没有类似的问题,每次代码运行到这里通过的都很快。
YOLOv5 🚀 2022-9-11 Python-3.8.0 torch-1.12.1+cu113 CUDA:0 (GeForce RTX 3050 Laptop GPU, 3902MiB)
Fusing layers...
Model summary: 213 layers, 7012822 parameters, 0 gradients