RuntimeError: CuDNN error: CUDNN_STATUS_SUCCESS

博客主要围绕Python中使用PyTorch或TensorFlow时遇到的CuDNN错误展开,提及了RuntimeError: CuDNN error: CUDNN_STATUS_SUCCESS和RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED两个错误,并给出了前者的解决方法,即在前面加入特定代码。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

问题一: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

评论 16
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值