InternalError (see above for traceback): Blas GEMM launch failed已解决

Tensorflow配置的环境是Win10 RTX2060显存,CUDA 9.0,cudnn7,tensorflow-gpu1.5。
运行程序出现InternalError (see above for traceback): Blas GEMM launch failed : a.shape=(1, 25088), b.shape=(25088, 4096), m=1, n=4096, k=25088错误,网上很多回答是关于tensorflow在调用GPU时的显存分配出现问题,给出的解决方法是重启电脑关闭多余进程或者动态分配显存。试了很多次都不行。因为同一个程序在之前的电脑同样的配置,只有1050Ti4G显存的情况下都可以运行,当显卡性能和显存提高时反而运行不了,改变显存分配方式显然是不对的。无意中看到https://www.jianshu.com/p/8f774b7866c7 的回答,说这是CUDA9与cudnn的不兼容问题,在CUDA的官网上给出了cuda9的补丁,这个补丁都是关于cuBLAS。
InternalError (see above for traceback): Blas GEMM launch failed已解决_第1张图片
依次下载和安装这四个补丁,即可解决这个问题。安装第一个补丁时出现闪退,删除文件夹C:\Program Files\NVIDIA Corporation\Installer2,再依次安装这四个补丁,可解决InternalError (see above for traceback): Blas GEMM launch failed。

你可能感兴趣的:(深度学习目标检测)