- 博客(2)
- 收藏
- 关注
RSS订阅原创 小白初学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
984
原创 小白初学YOLOv8报错：ValueError: not enough values to unpack (expected 3, got 0)
暂时还没找到很好的解决方法，看到网上有博客说是官方最新代码有误（我的日期为2023.10.24下载），等我回退版本后跑一下再来反馈。第一次使用yolov8跑出错了。
2023-10-25 10:08:01
609
3
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人