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)

你可能感兴趣的:(程序猿,Pytorch)