Pytorch解决 RuntimeError: CuDNN error: CUDNN_STATUS_SUCCESS

当我在Pytorch中使用RNN,并把其送到GPU上进行运算时,出现了:
RuntimeError: CuDNN error: CUDNN_STATUS_SUCCESS

在错误出现前,我的环境是:
OS:Ubuntu 16.0.4
GPU: NVIDIA GForce RTX 2080(查看GPU及驱动信息可使用命令:nvidia-smi)
CUDA:cuda9.0
cudnn:7.X for cuda9.0

在网上查了一下解决方法,其实遇到这种问题很简单,就是

1.更新cuda和cudnn
把你的cuda和cudnn换为9.2或者10.0或者你的GPU支持的cuda最低版本以上的版本;

2. 更新pytorch和torchvision版本
记住,在更新cuda和cudnn后,一定要把pytorch更新到适合你cuda的版本,比如 torch0.4.1 for cuda9.2。

以上2步做完后,基本就能解决这个问题了。参考的方法为:

https://discuss.pytorch.org/t/runtimeerror-cudnn-error-cudnn-status-success/28045)

  • 11
    点赞
  • 22
    收藏
    觉得还不错? 一键收藏
  • 14
    评论
评论 14
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值