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

【CUDA error: CUBLAS

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

【CUDA error: CUBLAS

CUDA error: CUBLAS_STATUS_EXECUTION_FAILED 解决方法

最近在训练StructureNet的时候出现了一个错误:

这个错误是在反向传播 backward() 那一步出现的,网上搜了很多相关资料,大部分说法是CUDA 和 PyTorch 版本不匹配造成的,但是换了几个 CUDA 版本还是会出现同样的问题。

最后在GitHub上面看到一个Issue,其中有一个人这样子回复:

看来有可能是 PyTorch 的问题,由于模型比较老,担心新版本会出现其他问题,于是装了一个 PyTorch 1.7 成功运行!

这里还要注意一下 torch-scatter 的问题,我在直接 pip install torch-scatter 的时候出现报错,查了一下发现需要手动安装对应版本的 torch-scatter (参考:https://blog.csdn.net/dream__1/article/details/122433061)

在 https://pytorch-geometric.com/whl/ 找到对应版本的 torch-scatter,复制链接, pip install 安装

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

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

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