环境windows10、tensorflow-gpu1.9.0、keras2.2.0、cuda9.0、cudnn7、显卡2080Ti
报错:InternalError: Blas GEMM launch failed : a.shape=(100, 784), b.shape=(784, 10), m=100, n=10...
我的先说段废话,我看了两天博客试遍了博客几乎所有方法,我的头很痛,周六日放假了还在想这个问题,最终皇天不负有心人,解决了,在不经意间没看到了一篇博客,但是还是不是很详细,让我找到了方向,加上自己一番摸索终于解决了,好了到这结束,下面是说问题,转载请注明出处,不容易,辛苦了几天,大家多多转载吧,留给有需要的人。
转载请备注出处谢谢了。。。。。。。。。。
解释:显卡性能和显存提高时反而运行不了,改变显存分配方式显然是不对的。无意中看到https://www.jianshu.com/p/8f774b7866c7 的回答,说这是CUDA9与cudnn的不兼容问题,在CUDA的官网上给出了cuda9的补丁,这个补丁都是关于cuBLAS
依次下载和安装这四个补丁,即可解决这个问题。安装第一个补丁时出现闪退,删除文件夹C:\Program Files\NVIDIA Corporation\Installer2,再依次安装这四个补丁,可解决InternalError (see above for traceback): Blas GEMM launch failed。
————————————————————截止
上面是我看到的别人博客说的,呵呵大,这就让我有了方向,头疼的是,说是补丁问题但是没有留补丁地址,头大,上cudn官网也没有找到对了,头大。。。。。。。。。。。
我的步骤是:删除文件夹C:\Program Files\NVIDIA Corporation\Installer2(我是改名了没有真删除,怕出事),再下载安装补丁
补丁在哪里,我上传到我的下载上了,可以自行下载,安装
补充:
如果上面还是解决不了问题,请仔细看cuda+cudnn安装文档的要求,特别是cudnn文件要放的位置,图像引擎版本要求,还有,就是补丁下载位置“我的下载”里面下最终版