自定义博客皮肤VIP专享

*博客头图:

格式为PNG、JPG,宽度*高度大于1920*100像素,不超过2MB,主视觉建议放在右侧,请参照线上博客头图

请上传大于1920*100像素的图片!

博客底图:

图片格式为PNG、JPG,不超过1MB,可上下左右平铺至整个背景

栏目图:

图片格式为PNG、JPG,图片宽度*高度为300*38像素,不超过0.5MB

主标题颜色:

RGB颜色,例如:#AFAFAF

Hover:

RGB颜色,例如:#AFAFAF

副标题颜色:

RGB颜色,例如:#AFAFAF

自定义博客皮肤

-+
  • 博客(2)
  • 问答 (1)
  • 收藏
  • 关注

原创 YOLOv8+DCNv3报错:RuntimeError: Not implemented on the CPU

使用的是yolov8 pytorch1.12.0+cuda11.3,出现了这种情况,已经设置了device使用GPU,训练时还是会提示不能使用CPU报错,根据网上。2.生成的全0张量和model都放入GPU运算。第一次遍历模型使用的def _predict_once方法中也加入self.model.cuda()之后就可以成功运行了!

2024-04-17 20:33:29 1243 5

原创 4090上训练yolov8出现 nvrtc: error:invalid value for --gpu-architecture(-arch)

由于4090不支持cuda11.3,所以换成cuda11.8。之前装的pytorch和cuda11.3。在4090上训练yolov8时发现。

2024-04-11 12:19:08 1027 1

空空如也

TA创建的收藏夹 TA关注的收藏夹

TA关注的人

提示
确定要删除当前文章?
取消 删除