error when training

[buaaswf]

File "train_FN.py", line 400, in main() File "train_FN.py", line 279, in main use_gt_boxes=args.use_gt_boxes) File "/home/swf/F-Net/models/HDN_v2/engines_v1.py", line 133, in test use_gt_boxes=use_gt_boxes) File "/home/swf/F-Net/models/HDN_v2/factorizable_network_v4.py", line 280, in evaluate object_result, predicate_result = self.forward_eval(im_data, im_info,) File "/home/swf/F-Net/models/HDN_v2/factorizable_network_v4.py", line 232, in forward_eval features, object_rois, _ = self.rpn(im_data, im_info) File "/home/swf/anaconda3/envs/f-net/lib/python2.7/site-packages/torch/nn/modules/module.py", line 489, in call result = self.forward(*input, **kwargs) File "/home/swf/F-Net/models/RPN/RPN.py", line 115, in forward mappings=self.opts['mappings']) File "/home/swf/F-Net/models/RPN/RPN.py", line 141, in proposal_layer _feat_stride, opts, anchor_scales, anchor_ratios, mappings) File "/home/swf/F-Net/lib/rpn_msr/proposal_layer.py", line 134, in proposal_layer keep = nms(np.hstack((proposals, scores)).astype(np.float32), nms_thres) File "/home/swf/F-Net/lib/fast_rcnn/nms_wrapper.py", line 26, in nms return nms_gpu(dets[:, :4], dets[:, 4], thresh).cpu().numpy() RuntimeError: Not compiled with GPU support (nms at /home/dell/F-Net/lib/layer_utils/csrc/nms.h:22) frame #0: c10::Error::Error(c10::SourceLocation, std::string const&) + 0x45 (0x7fc66e615cf5 in /home/swf/anaconda3/envs/f-net/lib/python2.7/site-packages/torch/lib/libc10.so)

[vishalrishu]

I'm getting same error. Please let me know if you get the solution to this query. Thanks @buaaswf

[FloatyDev]

has anyone found the solution to this?