一、原因:
这可能是因为您的显卡太旧或者计算压力对于您的显卡来说太重了。
就像在 CUDA 张量上进行多线程处理一样无法成功,有两种方法可供选择:
1. 不要使用多线程处理。将 DataLoader 的 num_worker 设置为零。
2. 改为 CPU 共享张量。确保您的自定义 DataSet 数据集返回CPU张量。
二、方法
在命令行后面添加worker=0
yolo task=detect mode=train model=yolov8n.pt data=E:\rebecca\yolov8-EL\datasets\eldata\ELdata.yaml epochs=100 epochs=300 batch=16 workers=0
参考链接:https://blog.csdn.net/u010440456/article/details/119187943