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

Google colaboratory cuda运行时错误(48):无法识别的错误代码,google-colaboratory,Google Colaboratory,我可以在T4 GPU上运行mmdetection代码,但在K80和P100 GPU上不工作。当我执行测试代码()时,我得到以下错误: # 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/faste

我可以在T4 GPU上运行mmdetection代码,但在K80和P100 GPU上不工作。当我执行测试代码()时,我得到以下错误:

# 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

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

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

所以代码对我有用

有关版本,您可以参考以下链接:


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

所以代码对我有用

有关版本,您可以参考以下链接:


我面临着同样的错误,你找到解决问题的方法了吗???我面临着同样的错误,你找到解决问题的方法了吗???