2021年11月5日
Pytorch 中RNN网络CUDNN_STATUS_INTERNAL_ERROR
摘要： 经验证，在config文件中的device设置下添加 torch.cuda.set_device(3)该问题应是因为cuda内部对于RNN的参数分配不当导致的。添加上面这一行的设置后即解决该问题。即改为如下设置： device = 'cuda:3' if torch.cuda.is_availabl 阅读全文
posted @ 2021-11-05 15:41 呦呦南山 阅读(190) 评论(0) 推荐(0) 编辑