CUDA error, my cuda version is 10.0, jump out error like this

mjq11302010044 / RRPN_pytorch

RRPN in pytorch 1.0 ----- Arbitrary-Oriented Scene Text Detection via Rotation Proposals

MIT License

262 stars 56 forks source link

CUDA error, my cuda version is 10.0, jump out error like this #53

Open oceanogeology opened 4 years ago

oceanogeology commented 4 years ago

File "../RRPN_pytorch/maskrcnn_benchmark/structures/bounding_box.py", line 197, in getitem bbox = RBoxList(self.bbox[item], self.size, self.mode) RuntimeError: CUDA error: invalid device function

mjq11302010044 commented 4 years ago

@oceanogeology Does the bug come out in single GPU mode?

oceanogeology commented 4 years ago

@mjq11302010044 I tried single GPU and multiple 4 GPU, error goes following:

File "../RRPN_pytorch/maskrcnn_benchmark/layers/rroi_align.py", line 30, in backward rois, con_idx_x, con_idx_y = ctx.saved_tensors ValueError: not enough values to unpack (expected 3, got 0)