RuntimeError: cuDNN error: CUDNN

编程入门 行业动态 更新时间:2024-10-22 11:25:35

RuntimeError: <a href=https://www.elefans.com/category/jswz/34/1768076.html style=cuDNN error: CUDNN"/>

RuntimeError: cuDNN error: CUDNN

RuntimeError: cuDNN error: CUDNN_STATUS_BAD_PARAM

这个错误是由于数据类型不一致造成的:

x_emb=self.gen_embedding(x) #[83, 8, 300] max_len, batch, dim
x_emb=self.dropout(x_emb)
x_emb = x_emb.to(torch.float32)
total_length = x_emb.size(0) #句子的最大长度
packed_emb=torch.nn.utils.rnn.pack_padded_sequence(x_emb, x_len) #[110, 300]即:
encoder_outputs, encoder_hidden = self.rnn(packed_emb) 

这里最后一句出错
因为需求是torch.float32,但是输入是torch.float64,所以会出现错误。
将x_emb转换为torch.float32类型就可以了:x_emb = x_emb.to(torch.float32)

更多推荐

RuntimeError: cuDNN error: CUDNN

本文发布于:2024-02-05 09:09:44,感谢您对本站的认可!
本文链接:https://www.elefans.com/category/jswz/34/1674019.html
版权声明:本站内容均来自互联网,仅供演示用,请勿用于商业和其他非法用途。如果侵犯了您的权益请与我们联系,我们将在24小时内删除。
本文标签:cuDNN   RuntimeError   CUDNN   error

发布评论

评论列表 (有 0 条评论)
草根站长

>www.elefans.com

编程频道|电子爱好者 - 技术资讯及电子产品介绍!