Meet Runtime:error When go to the step Demo and Test with pre-trained models
Seanlocked opened this issue · 0 comments
Seanlocked commented
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`