RuntimeError: CuDNN error: CUDNN_STATUS_SUCCESS

版權聲明：本文爲博主原創文章，未經博主允許不得轉載。
問題一：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