之前用的CPU，现在用GPU出现这样的问题，找到其他博主的额解决办法，例：
检查CUDA和pytorch是否兼容
返回True,兼容
应该是代码的问题，搜索到device
设置device='0'
开始训练
YOLOV5使用GPU训练出现CUDA和pytorch不兼容，但是输出是True
猜你喜欢
转载自blog.csdn.net/weixin_47037450/article/details/130468033
今日推荐
周排行