您的位置:首页 > 理论基础 > 计算机网络

RTX20系列显卡运行网络出现cublas runtime error: the GPU program failed to execute的解决办法

2020-02-01 22:48 2641 查看

RTX20系列显卡运行网络出现cublas runtime error: the GPU program failed to execute的解决办法

RTX20系列显卡运行网络出现cublas runtime error: the GPU program failed to execute的解决办法

做人脸相关深度学习时遇到了cublas runtime error: the GPU program failed to execute的问题。

配置

系统:win10
显卡:RTX2060 SUPER (6G)
CUDA:10.1, V10.1.243
pytorch版本:1.0.1
PyCharm:2019.2 x64

解决方法

主要是RTX20和pytorch以及CUDA的匹配版本问题,然后呢我是太懒每次都从pycharm来安装package但是pycharm只能更新pytorch本身导致了这个问题。

那么,从官方重装就好了。Pytorch官方链接

根据命令更新即可解决问题。

  • 点赞
  • 收藏
  • 分享
  • 文章举报
DJ-Jacob 发布了1 篇原创文章 · 获赞 0 · 访问量 353 私信 关注
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: 
相关文章推荐