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出现RuntimeError: cublas runtime error :cu:259问题,及解决方法_第1张图片

这个错误,在今天我用新的环境pytorch-1.1.0时遇到了;但我之前的环境pytorch-m不会出现这样的错误。于是我仔细查看了两个环境的区别,使用conda list,发现前者是:
pytorch出现RuntimeError: cublas runtime error :cu:259问题,及解决方法_第2张图片

后者是:
pytorch出现RuntimeError: cublas runtime error :cu:259问题,及解决方法_第3张图片

可以发现后者根本没有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的区别……

你可能感兴趣的:(问题解决,深度学习,pytorch)