ubuntu,tensorflow InternalError (see above for traceback): Blas GEMM launch failed解决方案

运行网络时遇到:

InternalError (see above for traceback): Blas GEMM launch failed : a.shape=(3, 2), b.shape=(2, 3), m=3, n=3, k=2

[[Node: MatMul = MatMul[T=DT_FLOAT, transpose_a=false, transpose_b=false, _device="/job:localhost/replica:0/task:0/device:GPU:0"](_arg_input_0_0/_1, Variable/read)]]

解决方法:

1.检查session关闭。

if 'session' in locals() and session is not None:
    print('Close interactive session')

    session.close()

2.重启机器(亲测有用)

你可能感兴趣的:(code)