-
Notifications
You must be signed in to change notification settings - Fork 2.3k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Training issue #414
Comments
NMS seems to be compiled without GPU. Try removing the |
Yes, it's seem like is nvidia's problem. Thank you! |
I also encountered the same problem. How do you solve the problem specifically? |
1.install cuda和cudnn correctly (be careful about their version) |
Hello, Can you talk about how to ensure my CUDA/CUDNN in my path? thanks~ |
Hello, Have you done this step? Appreciate. |
Sorry, it's a long time ago, I forget My steps. And I remember I didn't fix this problem |
I‘ve done it.Thanks for reply anyway
发自我的iPhone
…------------------ Original ------------------
From: Kim Yang <[email protected]>
Date: Sun,Jan 10,2021 0:26 AM
To: jwyang/faster-rcnn.pytorch <[email protected]>
Cc: Lyz <[email protected]>, Comment <[email protected]>
Subject: Re: [jwyang/faster-rcnn.pytorch] Training issue (#414)
Hello, Can you talk about how to ensure my CUDA/CUDNN in my path? thanks~
Hello, Have you done this step? Appreciate.
Sorry, it's a long time ago, I forget My steps. And I remember I didn't fix this problem
—
You are receiving this because you commented.
Reply to this email directly, view it on GitHub, or unsubscribe.
|
I‘ve solved it. Thank you very much |
Preparing training data...
done
before filtering, there are 10022 images...
after filtering, there are 10022 images...
10022 roidb entries
Loading pretrained weights from data/pretrained_model/vgg16_caffe.pth
Traceback (most recent call last):
File "trainval_net.py", line 320, in
rois_label = fasterRCNN(im_data, im_info, gt_boxes, num_boxes)
File "/home/janvin/anaconda3/envs/python36/lib/python3.6/site-packages/torch/nn/modules/module.py", line 489, in call
result = self.forward(*input, **kwargs)
File "/home/janvin/faster-rcnn.pytorch/lib/model/faster_rcnn/faster_rcnn.py", line 54, in forward
rois, rpn_loss_cls, rpn_loss_bbox = self.RCNN_rpn(base_feat, im_info, gt_boxes, num_boxes)
File "/home/janvin/anaconda3/envs/python36/lib/python3.6/site-packages/torch/nn/modules/module.py", line 489, in call
result = self.forward(*input, **kwargs)
File "/home/janvin/faster-rcnn.pytorch/lib/model/rpn/rpn.py", line 78, in forward
im_info, cfg_key))
File "/home/janvin/anaconda3/envs/python36/lib/python3.6/site-packages/torch/nn/modules/module.py", line 489, in call
result = self.forward(*input, **kwargs)
File "/home/janvin/faster-rcnn.pytorch/lib/model/rpn/proposal_layer.py", line 147, in forward
keep_idx_i = nms(proposals_single, scores_single.squeeze(1), nms_thresh)
RuntimeError: Not compiled with GPU support (nms at /home/janvin/faster-rcnn.pytorch/lib/model/csrc/nms.h:22)
frame #0: c10::Error::Error(c10::SourceLocation, std::string const&) + 0x45 (0x7fc141178cc5 in /home/janvin/anaconda3/envs/python36/lib/python3.6/site-packages/torch/lib/libc10.so)
frame #1: nms(at::Tensor const&, at::Tensor const&, float) + 0xdc (0x7fc13ceae48c in /home/janvin/faster-rcnn.pytorch/lib/model/_C.cpython-36m-x86_64-linux-gnu.so)
frame #2: + 0x148e8 (0x7fc13ceba8e8 in /home/janvin/faster-rcnn.pytorch/lib/model/_C.cpython-36m-x86_64-linux-gnu.so)
frame #3: + 0x11f35 (0x7fc13ceb7f35 in /home/janvin/faster-rcnn.pytorch/lib/model/_C.cpython-36m-x86_64-linux-gnu.so)
frame #56: __libc_start_main + 0xe7 (0x7fc192bcab97 in /lib/x86_64-linux-gnu/libc.so.6)
what's the problem is ?
The text was updated successfully, but these errors were encountered: