Skip to content
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

Meet Runtime:error When go to the step Demo and Test with pre-trained models #153

Open
Seanlocked opened this issue Oct 11, 2019 · 0 comments

Comments

@Seanlocked
Copy link

I met the Error above
RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED

My Ubuntu is 16.04
My pytorch version is 1.1.0
My CUDA Version 10.0.130
My CUDNN_MAJOR 7

`(base) root@809026aa6fb2:~/pytorch-faster-rcnn# CUDA_VISIBLE_DEVICES=0 ./tools/demo.py

Loaded network output/res101/voc_2007_trainval+voc_2012_trainval/default/res101_faster_rcnn_iter_110000.pth

Demo for data/demo/000456.jpg
Traceback (most recent call last):
  File "./tools/demo.py", line 179, in <module>
    demo(net, im_name)
  File "./tools/demo.py", line 96, in demo
    scores, boxes = im_detect(net, im)
  File "/root/pytorch-faster-rcnn/tools/../lib/model/test.py", line 107, in im_detect
    blobs['im_info'])
  File "/root/pytorch-faster-rcnn/tools/../lib/nets/network.py", line 450, in test_image
    self.forward(image, im_info, None, mode='TEST')
  File "/root/pytorch-faster-rcnn/tools/../lib/nets/network.py", line 409, in forward
    rois, cls_prob, bbox_pred = self._predict()
  File "/root/pytorch-faster-rcnn/tools/../lib/nets/network.py", line 387, in _predict
    fc7 = self._head_to_tail(pool5)
  File "/root/pytorch-faster-rcnn/tools/../lib/nets/resnet_v1.py", line 120, in _head_to_tail
    fc7 = self.resnet.layer4(pool5).mean(3).mean(
  File "/root/anaconda3/lib/python3.7/site-packages/torch/nn/modules/module.py", line 493, in __call__
    result = self.forward(*input, **kwargs)
  File "/root/anaconda3/lib/python3.7/site-packages/torch/nn/modules/container.py", line 92, in forward
    input = module(input)
  File "/root/anaconda3/lib/python3.7/site-packages/torch/nn/modules/module.py", line 493, in __call__
    result = self.forward(*input, **kwargs)
  File "/root/anaconda3/lib/python3.7/site-packages/torchvision/models/resnet.py", line 98, in forward
    out = self.conv2(out)
  File "/root/anaconda3/lib/python3.7/site-packages/torch/nn/modules/module.py", line 493, in __call__
    result = self.forward(*input, **kwargs)
  File "/root/anaconda3/lib/python3.7/site-packages/torch/nn/modules/conv.py", line 338, in forward
    self.padding, self.dilation, self.groups)
RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED`
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant