facebookresearch / maskrcnn-benchmark

Fast, modular reference implementation of Instance Segmentation and Object Detection algorithms in PyTorch.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Not compiled with GPU support

AnaRhisT94 opened this issue · comments

❓ Questions and Help

Hi,
I'm having issues regarding roi_align.py, it always throws
Not compiled with GPU support in the following line:
output = _C.roi_align_forward(input, roi, spatial_scale, output_size[0], output_size[1], sampling_ratio)

Any help is highly appreciated.