MaskDINO icon indicating copy to clipboard operation
MaskDINO copied to clipboard

Published 20 hours ago • verified publisher icon IDEA-Research

Cannot specify GPU when carrying out inference with mask DINO

Open keithkctse opened this issue 1 year ago • 4 comments

When I set cfg.MODEL.DEVICE = 'cuda:1' ,it says illegal memory access. The problem does not exist when i set cfg.MODEL.DEVICE = 'cuda:0', how to solve this?

image

keithkctse avatar Mar 26 '23 09:03 keithkctse