栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

RuntimeError: cuDNN error: CUDNN

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

RuntimeError: cuDNN error: CUDNN

加载预训练模型并增加一层nn.Embedding层,运行代码时,抛出RuntimeError: cuDNN error: CUDNN_STATUS_NOT_INITIALIZED的错误。这种错误一般情况下是由于pytorch和CUDA版本不对应引起的,但是预训练模型在同样的环境下能跑通。故,排除版本不匹配问题。经检查,原因是自己增加的nn.Embedding(vocab_size, dim)中的vocab_size=49417设置错误。我的词典从0到49417共有49418个单词,而我一开始设置的是49417,查询表大小错误,将其更改为size应该设置为49418后,错误解决。

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/972117.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号