tensorflow2.0 InternalError: 2 root error(s) found. Internal: Blas GEMM launch failed解决办法

在代码最前面加入代码

注意 ，这里为tensorflow2.0版本，与第1.0有差距。
 

import os

os.environ["CUDA_VISIBLE_DEVICES"] = "0"

#原版 config = tf.ConfigProto(allow_soft_placement=True)
config = tf.compat.v1.ConfigProto(allow_soft_placement=True)
#原版 sess = tf.Session(config=config)
gpu_options = tf.compat.v1.GPUOptions(per_process_gpu_memory_fraction = 0.5)

config.gpu_options.allow_growth = True

sess =tf.compat.v1.Session(config=config)  #注意 ，这里为tensorflow2.0版本，与第1.0有差距。