ошибка выполнения cuda (48): нераспознанный код ошибки

Question

Я могу запустить mmdetection code на T4 GPU, но он не работает на 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

Кто-нибудь сталкивался с этой проблемой? Вы можете помочь мне решить это?