ruotianluo/pytorch-faster-rcnn

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

Seanlocked opened this issue · 0 comments

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`