出现Blas GEMM launch failed报错的原因是tensorflow在调用GPU时的显存分配出现问题,tensorflow默认申请可使用的全部显存,当tensorflow程序运行会话却没有关闭会话释放资源的时候,就会出现此错误。
解决方法
在运行Session时将:
with tf.Session() as sess:
替换为:
gpu_options = tf.GPUOptions(allow_growth=True)
with tf.Session(config=tf.ConfigProto(gpu_options=gpu_options)) as sess:
即将显存分配方式更改为按需分配即可。