【踩坑】RuntimeError: CUDA error: CUBLAS_STATUS_NOT_INITIALIZED when calling `cublasCreate(handle)`

今天NLP了吗

50667人浏览 · 2021-04-23 11:17:13

今天NLP了吗 · 2021-04-23 11:17:13 发布

在GPU运行报错

RuntimeError: CUDA error: CUBLAS_STATUS_NOT_INITIALIZED when calling `cublasCreate(handle)`

在CPU运行报错

index out of range in self

看了一些回答说batch size太大，或者CUDA版本和torch不匹配，尝试无果。

有可能是embedding的问题
这个答案对我也没效果

最后把网络的各个shape打印出来，发现其实是数据shape不匹配。

有问题先去看数据、embedding、网络维度！！！

智能机器人开发者大赛社区

「智能机器人开发者大赛」官方平台，致力于为开发者和参赛选手提供赛事技术指导、行业标准解读及团队实战案例解析;聚焦智能机器人开发全栈技术闭环，助力开发者攻克技术瓶颈，促进软硬件集成、场景应用及商业化落地的深度研讨。加入智能机器人开发者社区iRobot Developer，与全球极客并肩突破技术边界，定义机器人开发的未来范式！

更多推荐