自定义博客皮肤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)
  • 收藏
  • 关注

原创 小白初学YOLOv8 NotImplementedError: Could not run ‘torchvision::nms‘ with arguments from the ‘CUDA‘ back

在本地跑yolov8模型遇见的一个报错,查了网上很多说法是torch版本和cuda版本的问题,目前我的GPU cuda版本是12.1,之前用的torch 12.1和torchvision 0.16+cpu(当时没意识到torchvision版本不支持cuda)会报错。后续我降低了torch和torchvision的版本,分别是11.1和0.16+cu112,但还是会报错。在此记录一下,后续找到解决方法会更新。

2023-10-25 22:07:57 855

原创 小白初学YOLOv8报错:ValueError: not enough values to unpack (expected 3, got 0)

暂时还没找到很好的解决方法,看到网上有博客说是官方最新代码有误(我的日期为2023.10.24下载),等我回退版本后跑一下再来反馈。第一次使用yolov8跑出错了。

2023-10-25 10:08:01 495 3

空空如也

空空如也

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

TA关注的人

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