在本地跑yolov8模型遇见的一个报错,查了网上很多说法是torch版本和cuda版本的问题,目前我的GPU cuda版本是12.1,之前用的torch 12.1和torchvision 0.16+cpu(当时没意识到torchvision版本不支持cuda)会报错。
后续我降低了torch和torchvision的版本,分别是11.1和0.16+cu112,但还是会报错。在此记录一下,后续找到解决方法会更新。
后续终于解决了,降低版本后需要重启电脑,虽然不知道为什么重启电脑才能生效,但总归是能跑起来了