运行时错误:CUDNN_STATUS_INTERNAL_ERROR。

4
在ubuntu14.04系统上,我使用带有cudnn的pytorch时遇到了问题。具体问题如下:
Traceback (most recent call last):
  File "main.py", line 58, in <module>
    test_detect(test_loader, nod_net, get_pbb, bbox_result_path,config1,n_gpu=config_submit['n_gpu'])
  File "/home/ubuntu/nndl/DSB2017/test_detect.py", line 52, in test_detect
    output = net(input,inputcoord)
  File "/home/ubuntu/anaconda2/lib/python2.7/site-packages/torch/nn/modules/module.py", line 252, in __call__
    result = self.forward(*input, **kwargs)
  File "/home/ubuntu/anaconda2/lib/python2.7/site-packages/torch/nn/parallel/data_parallel.py", line 58, in forward
    return self.module(*inputs[0], **kwargs[0])
  File "/home/ubuntu/anaconda2/lib/python2.7/site-packages/torch/nn/modules/module.py", line 252, in __call__
    result = self.forward(*input, **kwargs)
  File "/home/ubuntu/nndl/DSB2017/net_detector.py", line 102, in forward
    out = self.preBlock(x)#16
  File "/home/ubuntu/anaconda2/lib/python2.7/site-packages/torch/nn/modules/module.py", line 252, in __call__
    result = self.forward(*input, **kwargs)
  File "/home/ubuntu/anaconda2/lib/python2.7/site-packages/torch/nn/modules/container.py", line 67, in forward
    input = module(input)
  File "/home/ubuntu/anaconda2/lib/python2.7/site-packages/torch/nn/modules/module.py", line 252, in __call__
    result = self.forward(*input, **kwargs)
  File "/home/ubuntu/anaconda2/lib/python2.7/site-packages/torch/nn/modules/conv.py", line 351, in forward
    self.padding, self.dilation, self.groups)
  File "/home/ubuntu/anaconda2/lib/python2.7/site-packages/torch/nn/functional.py", line 119, in conv3d
    return f(input, weight, bias)
RuntimeError: CUDNN_STATUS_INTERNAL_ERROR

我已经谷歌搜索了几个小时,但仍然感到非常困惑。是什么导致了这种情况?


显卡是GTX1080Ti。CUDA8.0,cuDNN5.1。 - xu wang
1个回答

9

我在 Ubuntu 16.04 上遇到了这个问题并解决了它。我的解决方案是运行:

sudo rm -rf ~/.nv

然后重新启动。


1
如果我无法使用“sudo”命令怎么办? - Shai
1
@Shai 前往 home 文件夹,按下 Ctrl+H 显示隐藏文件夹,然后删除 .nv 文件夹。我想这应该也可以起作用。 - Peter

网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接