在跑程序时,出现下面的错误:
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,发现前者是:
后者是:
可以发现后者根本没有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的区别……