cuda运行时错误(48)：无法识别的错误代码

Question

我可以在T4图形处理器上运行mmdetection代码，但它在K80和P100 GPU.When上不起作用。我执行测试代码()，得到以下错误：

# This is the test code
from mmdet.apis import init_detector, inference_detector, show_result
config_file = 'configs/my_faster_rcnn_r50_fpn_1x.py'
checkpoint_file = './work_dirs/faster_rcnn_r50_fpn_1x/epoch_9.pth'
model = init_detector(config_file, checkpoint_file, device='cuda:0')
img = 'demo.jpg'
result = inference_detector(model, img)
show_result(img, result, model.CLASSES)

以下是错误输出：

# errpr log
/content/drive/My Drive/model_train/10_酒瓶检测/mmdetection/mmdet/ops/roi_align/roi_align.py in forward(ctx, features, rois, out_size, spatial_scale, sample_num)
     25             # print('进入')
     26             roi_align_cuda.forward(features, rois, out_h, out_w, spatial_scale,
---> 27                                    sample_num, output)
     28         else:
     29             print('出错')

RuntimeError: cuda runtime error (48) : unrecognized error code at mmdet/ops/roi_align/src/roi_align_kernel.cu:140

有没有人遇到过这个问题？你能帮我解决吗？

Answer 1

我遇到了同样的问题，在分析这个问题时，这个错误似乎是由于Pytorch与CUDA版本不兼容造成的。

所以代码对我来说是有效的。

您可以参考下面的链接了解版本：

https://pytorch.org/get-started/previous-versions/