search

xingyizhou / ExtremeNet

Bottom-up Object Detection by Grouping Extreme and Center Points
BSD 3-Clause "New" or "Revised" License
1.03k stars 172 forks source link

RuntimeError: CUDNN_STATUS_EXECUTION_FAILED heeeelp please :( #49

Open pcuevas-ch opened 1 year ago

pcuevas-ch commented 1 year ago

Hello ,

I'm trying to run demo.py but it gives me this error. The environment is given mainly by:

This is the whole error:

Traceback (most recent call last): File "demo.py", line 189, in kernel=nms_kernel, debug=True) File "demo.py", line 86, in kp_decode scores_thresh=scores_thresh, center_thresh=center_thresh, debug=debug) File "/home/DIINF/pcuevas/ExtremeNet/nnet/py_factory.py", line 99, in test return self.model(*xs, kwargs) File "/home/DIINF/pcuevas/anaconda3/envs/CornerNet/lib/python3.6/site-packages/torch/nn/modules/module.py", line 491, in call result = self.forward(*input, *kwargs) File "/home/DIINF/pcuevas/ExtremeNet/nnet/py_factory.py", line 31, in forward return self.module(xs, kwargs) File "/home/DIINF/pcuevas/anaconda3/envs/CornerNet/lib/python3.6/site-packages/torch/nn/modules/module.py", line 491, in call result = self.forward(*input, kwargs) File "/home/DIINF/pcuevas/ExtremeNet/models/py_utils/exkp.py", line 267, in forward return self._test(*xs, *kwargs) File "/home/DIINF/pcuevas/ExtremeNet/models/py_utils/exkp.py", line 226, in _test inter = self.pre(image) File "/home/DIINF/pcuevas/anaconda3/envs/CornerNet/lib/python3.6/site-packages/torch/nn/modules/module.py", line 491, in call result = self.forward(input, kwargs) File "/home/DIINF/pcuevas/anaconda3/envs/CornerNet/lib/python3.6/site-packages/torch/nn/modules/container.py", line 91, in forward input = module(input) File "/home/DIINF/pcuevas/anaconda3/envs/CornerNet/lib/python3.6/site-packages/torch/nn/modules/module.py", line 491, in call result = self.forward(*input, *kwargs) File "/home/DIINF/pcuevas/ExtremeNet/models/py_utils/utils.py", line 14, in forward conv = self.conv(x) File "/home/DIINF/pcuevas/anaconda3/envs/CornerNet/lib/python3.6/site-packages/torch/nn/modules/module.py", line 491, in call result = self.forward(input, **kwargs) File "/home/DIINF/pcuevas/anaconda3/envs/CornerNet/lib/python3.6/site-packages/torch/nn/modules/conv.py", line 301, in forward self.padding, self.dilation, self.groups) RuntimeError: CUDNN_STATUS_EXECUTION_FAILED

Help me please! :( Anyone can help me to debug it?

Thanks in advance!