求助！train.py报错！

bubbliiiing / yolov7-pytorch

这是一个yolov7的库，可以用于训练自己的数据集。

GNU General Public License v3.0

861 stars 150 forks source link

Open MaXinying123 opened 2 years ago

MaXinying123 commented 2 years ago

求助！train.py始终跑不起来，报错内容为“RuntimeError: Expected a 'cpu' device type for generator but found 'cuda'”，请问大家有遇到类似问题吗，感谢指点！

bubbliiiing commented 1 year ago

这是，什么版本呢

OrionZJ commented 1 year ago

同上，一样的问题

MaXinying123 commented 1 year ago

您好，您的邮件我已收到，谢谢配合。——马欣莹