Pytorch 中RNN网络CUDNN_STATUS_INTERNAL_ERROR

经验证，在config文件中的device设置下添加

torch.cuda.set_device(3)


该问题应是因为cuda内部对于RNN的参数分配不当导致的。添加上面这一行的设置后即解决该问题。

即改为如下设置：

device = 'cuda:3' if torch.cuda.is_available() else 'cpu'
torch.cuda.set_device(3)

以上是针对pytorch的设置，针对tensorflow，可如下设置：

import os
os.environ['CUDA_ENABLE_DEVICES'] = '0'