pytorch出现RuntimeError: cublas runtime error :cu:259问题,及解决方法

在跑程序时,出现下面的错误:
RuntimeError: cublas runtime error : the GPU program failed to execute at C:/w/1/s/tmp_conda_3.6_041836/conda/conda-bld/pytorch_1556684464974/work/aten/src/THC/THCBlas.cu:259

在这里插入图片描述

这个错误,在今天我用新的环境pytorch-1.1.0时遇到了;但我之前的环境pytorch-m不会出现这样的错误。于是我仔细查看了两个环境的区别,使用conda list,发现前者是:
在这里插入图片描述

后者是:
Z3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl80MTc4MTEyMQ==,size_16,color_FFFFFF,t_70#pic_center)

可以发现后者根本没有pytorch,而是torch;而且下载源也不一样:另一方面前者是用官网的语句,用conda下载的,后者是在清华大学镜像网站中下载的。因此出现这个问题,很有可能是pytorch的来源不同。
因为要使用1.1.0版本,我又去清华大学镜像网站下了1.1.0的pytorch。配置好后,居然就没有这个错误了!

补充:
之前也出现过RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED这个错误。许多博客都在说是pytorch、cuda等版本对应的问题,但我的环境应该没有问题,故使用:
torch.backends.cudnn.enabled = False
强行让代码在cpu上面跑。
可是在安装了清华大学镜像网站下了1.1.0的pytorch后,这个错误神奇地消失了!!!
这个难以解释,有知道的大佬麻烦解释一下两个来源的pytorch的区别……

  • 7
    点赞
  • 19
    收藏
    觉得还不错? 一键收藏
  • 13
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 13
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值