pytorch RuntimeError: CuDNN error: CUDNN_STATUS_SUCCESS 如何解决
昨天在做LSTM神经网络预测的时候遇到了这个问题。报错信息是RuntimeError: CuDNN error: CUDNN_STATUS_SUCCESS。看起来好像是因为GPU加速模块的问题,但是换一个数据发现运行并没有什么问题,在网上查了好久也没找到问题的原因。其实是因为数据类型不符合。因为我当时选择的GPU加速训练,但是输入数据类型是float64的,改成torch.float32就好了!
呀哈~