RuntimeError: nms_impl: implementation for device cuda:0 not found.

阿基米德の支点

已于 2023-12-08 09:50:40 修改

阅读量3.8k

点赞数 11

分类专栏： debug 文章标签：深度学习人工智能 pytorch

于 2023-12-07 20:38:05 首次发布

本文链接：https://blog.csdn.net/seven_hells/article/details/134864191

版权

debug 专栏收录该内容

1 篇文章

订阅专栏

文章描述了在运行mmdetectionv3.2.0中FasterR-CNN模型时遇到的CUDA错误，由于找不到设备。作者通过检查环境确认GPU可用并发现版本匹配，但问题在于mmcv库。解决方案是卸载并重新安装mmcv。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

2023年12月7日
在跑mmdetection（v3.2.0）的faster RCNN模型时报错：

from mmdet.apis import DetInferencer

inferencer = DetInferencer(model='faster-rcnn_r50_fpn_1x_coco', device='cuda:0')

# print(inferencer.cfg)
# img_path = 'img'
inferencer(inputs='img',
           out_dir='faster_rcnn/results',
           return_vis=True,
           no_save_pred=False,
           batch_size=8
           )
print('done')

错误信息如标题所示，提示找不到显卡：
RuntimeError: nms_impl: implementation for device cuda:0 not found.

在该虚拟环境中运行代码检查是否能调用cuda：

import torch

print("是否可用：", torch.cuda.is_available())        # 查看GPU是否可用
print("GPU数量：", torch.cuda.device_count())        # 查看GPU数量
print("torch方法查看CUDA版本：", torch.version.cuda)  # torch方法查看CUDA版本
index = torch.cuda.current_device()
print("GPU索引号：", index)    # 查看GPU索引号
print("GPU名称：", torch.cuda.get_device_name(index))    # 根据索引号得到GPU名称

输出表示一切正常：

是否可用： True
GPU数量： 1
torch方法查看CUDA版本： 12.1
GPU索引号： 0
GPU名称： NVIDIA GeForce RTX 4070 Laptop GPU

检查torch，torchvision，torchaudio各包版本也都对得上（Python 3.8.18）：

!pip list

torch                     2.1.0+cu121       
torchaudio                2.1.0+cu121       
torchvision               0.16.0+cu121

解决方法——重装mmcv：

参考此篇博文的评论区：
https://blog.csdn.net/NCEPUDJ/article/details/121882436

mim uninstall mmcv
mim install mmcv

这里使用mim是因为mmdetection的官方文档里的教程：
https://mmdetection.readthedocs.io/zh-cn/latest/get_started.html
在这里插入图片描述