RuntimeError: CUDA error: CUBLAS_STATUS_INTERNAL_ERROR when calling `cublasCreate(handle)`

##RuntimeError: CUDA error: CUBLAS_STATUS_INTERNAL_ERROR when calling cublasCreate(handle)

最近在使用pytorch训练神经网络时出现了标题所示的错误,产生了很大的疑惑,因为在一个月前运行是不存在任何问题的,但是现在却出问题了,所以四处寻找各种解决方案,经过一整天的努力和尝试终于解决了这个问题。接下来把尝试过的方法分享给大家,其中最后一种方案是我适用的:

方法一

https://blog.csdn.net/xiaoxiaowantong/article/details/109689146

方法二

https://blog.csdn.net/weixin_45650259/article/details/109313193

方法三

https://blog.csdn.net/hunterflyy/article/details/108369274

方法四

在尝试以上方法没有成功解决后,我重新配置了一下环境仍然显示同样的错误,最后发现问题出现在pytorch和cuda的版本匹配上,我使用的pytorch版本为1.8.0,cuda版本为10.2,因此需要将pytorch版本降级为1.7.1,降级方法如下:

pip uninstall torch
pip install torch==1.7.1

最后建议大家再安装各种包时最好加上版本,以免带来不必要的麻烦。

你可能感兴趣的:(神经网络和深度学习,深度学习,python,机器学习,人工智能,cuda)