RTX20系列显卡运行网络出现cublas runtime error: the GPU program failed to execute的解决办法

RTX20系列显卡运行网络出现cublas runtime error: the GPU program failed to execute的解决办法

RTX20系列显卡运行网络出现cublas runtime error: the GPU program failed to execute的解决办法

做人脸相关深度学习时遇到了cublas runtime error: the GPU program failed to execute的问题。

配置

系统：win10
显卡：RTX2060 SUPER (6G)
CUDA：10.1, V10.1.243
pytorch版本：1.0.1
PyCharm：2019.2 x64

解决方法

主要是RTX20和pytorch以及CUDA的匹配版本问题，然后呢我是太懒每次都从pycharm来安装package但是pycharm只能更新pytorch本身导致了这个问题。

那么，从官方重装就好了。Pytorch官方链接

根据命令更新即可解决问题。