You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
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`
The text was updated successfully, but these errors were encountered:
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
The text was updated successfully, but these errors were encountered: