版权声明:本文为博主原创文章,未经博主允许不得转载。
问题一:RuntimeError: CuDNN error: CUDNN_STATUS_SUCCESS
File "C:\Users\tyy\AppData\Local\Continuum\Anaconda3\lib\site-packages\torch\nn\modules\module.py", line 477, in __call__
result = self.forward(*input, **kwargs)
File "C:\Users\tyy\AppData\Local\Continuum\Anaconda3\lib\site-packages\torch\nn\modules\conv.py", line 301, in forward
self.padding, self.dilation, self.groups)
RuntimeError: CuDNN error: CUDNN_STATUS_SUCCESS
解决方法:
在前面加入以下代码:
cudnn.benchmark = True
问题二:RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED