search

facebookresearch / detectron2

Detectron2 is a platform for object detection, segmentation and other visual recognition tasks.
https://detectron2.readthedocs.io/en/latest/
Apache License 2.0
30.02k stars 7.41k forks source link

Not compiled with GPU support #128

Closed jiushishuai88 closed 4 years ago

jiushishuai88 commented 4 years ago

(base) ai@ai-Precision-Tower-5810:~/detectron2$ python demo/demo.py --config-file configs/COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_3x.yaml --input input1.jpg input2.jpg --opts MODEL.WEIGHTS detectron2://COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_3x/137849600/model_final_f10217.pkl [10/21 17:15:38 detectron2]: Arguments: Namespace(confidence_threshold=0.5, config_file='configs/COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_3x.yaml', input=['input1.jpg', 'input2.jpg'], opts=['MODEL.WEIGHTS', 'detectron2://COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_3x/137849600/model_final_f10217.pkl'], output=None, video_input=None, webcam=False) WARNING [10/21 17:15:38 d2.config.compat]: Config 'configs/COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_3x.yaml' has no VERSION. Assuming it to be compatible with latest v2. Traceback (most recent call last): File "demo/demo.py", line 83, in predictions, visualized_output = demo.run_on_image(img) File "/home/ai/detectron2/demo/predictor.py", line 48, in run_on_image predictions = self.predictor(image) File "/home/ai/anaconda3/lib/python3.7/site-packages/torch/autograd/grad_mode.py", line 49, in decorate_no_grad return func(*args, kwargs) File "/home/ai/detectron2/detectron2/engine/defaults.py", line 177, in call predictions = self.model([inputs])[0] File "/home/ai/anaconda3/lib/python3.7/site-packages/torch/nn/modules/module.py", line 541, in call result = self.forward(*input, *kwargs) File "/home/ai/detectron2/detectron2/modeling/meta_arch/rcnn.py", line 66, in forward return self.inference(batched_inputs) File "/home/ai/detectron2/detectron2/modeling/metaarch/rcnn.py", line 124, in inference results, = self.roi_heads(images, features, proposals, None) File "/home/ai/anaconda3/lib/python3.7/site-packages/torch/nn/modules/module.py", line 541, in call result = self.forward(input, kwargs) File "/home/ai/detectron2/detectron2/modeling/roi_heads/roi_heads.py", line 564, in forward pred_instances = self._forward_box(features_list, proposals) File "/home/ai/detectron2/detectron2/modeling/roi_heads/roi_heads.py", line 611, in _forward_box box_features = self.box_pooler(features, [x.proposal_boxes for x in proposals]) File "/home/ai/anaconda3/lib/python3.7/site-packages/torch/nn/modules/module.py", line 541, in call result = self.forward(*input, *kwargs) File "/home/ai/detectron2/detectron2/modeling/poolers.py", line 195, in forward output[inds] = pooler(x_level, pooler_fmt_boxes_level) File "/home/ai/anaconda3/lib/python3.7/site-packages/torch/nn/modules/module.py", line 541, in call result = self.forward(input, **kwargs) File "/home/ai/detectron2/detectron2/layers/roi_align.py", line 95, in forward input, rois, self.output_size, self.spatial_scale, self.sampling_ratio, self.aligned File "/home/ai/detectron2/detectron2/layers/roi_align.py", line 20, in forward input, roi, spatial_scale, output_size[0], output_size[1], sampling_ratio, aligned RuntimeError: Not compiled with GPU support (ROIAlign_forward at /home/ai/detectron2/detectron2/layers/csrc/ROIAlign/ROIAlign.h:73) frame #0: c10::Error::Error(c10::SourceLocation, std::string const&) + 0x47 (0x7f66a1003687 in /home/ai/anaconda3/lib/python3.7/site-packages/torch/lib/libc10.so) frame #1: detectron2::ROIAlign_forward(at::Tensor const&, at::Tensor const&, float, int, int, int, bool) + 0x171 (0x7f6685bee4c1 in /home/ai/detectron2/detectron2/_C.cpython-37m-x86_64-linux-gnu.so) frame #2: + 0x215fe (0x7f6685bfe5fe in /home/ai/detectron2/detectron2/_C.cpython-37m-x86_64-linux-gnu.so) frame #3: + 0x1c3e0 (0x7f6685bf93e0 in /home/ai/detectron2/detectron2/_C.cpython-37m-x86_64-linux-gnu.so)

Environment

sys.platform linux Python 3.7.3 (default, Mar 27 2019, 22:11:17) [GCC 7.3.0] Numpy 1.16.4 Detectron2 Compiler GCC 7.4 Detectron2 CUDA Compiler not available DETECTRON2_ENV_MODULE PyTorch 1.3.0 PyTorch Debug Build False torchvision 0.4.1a0+d94043a CUDA available True GPU 0,1 GeForce GTX 1080 Ti CUDA_HOME None Pillow 6.1.0 cv2 3.4.2

PyTorch built with:

ppwwyyxx commented 4 years ago

Duplicate of #55