Pytorch cuDNN错误:cuDNN_STATUS_BAD_PARAM。有人能解释我为什么会出现此错误以及如何更正它吗?

Pytorch cuDNN错误:cuDNN_STATUS_BAD_PARAM。有人能解释我为什么会出现此错误以及如何更正它吗?,pytorch,cudnn,Pytorch,Cudnn,我试图使用Pytorch实现字符LSTM。但是我得到了cudnn_status_bad_params errors。这是训练循环。我在line output=model(input_seq)上得到了error 我得到了同样的错误,如果你切换到CPU,你会得到一个更好的错误描述。在我的例子中,问题在于我给网络的输入类型。我在发送我猜是长的,而模型需要浮动。我做了以下更改,代码运行正常。基本上,切换到cpu可以提供更好的错误描述 input_seq = input_seq.float().cuda(

我试图使用Pytorch实现字符LSTM。但是我得到了cudnn_status_bad_params errors。这是训练循环。我在line output=model(input_seq)上得到了error


我得到了同样的错误,如果你切换到CPU,你会得到一个更好的错误描述。在我的例子中,问题在于我给网络的输入类型。我在发送我猜是
长的
,而模型需要
浮动
。我做了以下更改,代码运行正常。基本上,切换到cpu可以提供更好的错误描述

input_seq = input_seq.float().cuda()

我遇到了同样的错误

您应该将输入类型从
float64
更改为
float32
,这意味着您应该键入:

input_seq = input_seq.float()

“如果你切换到CPU,你会得到一个更好的错误描述”绝对黄金这是一个答案的准确副本,已经发布了一年前你的。对不起,请指出我复制的来源,如果你认为它不是一个来源!!!此外,我还提到了我从哪里得到的解决方案。
input_seq = input_seq.float()