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

2023-11-06

在跑程序时,出现下面的错误:
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的区别……

本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)

pytorch出现RuntimeError: cublas runtime error :cu:259问题,及解决方法 的相关文章

随机推荐