在代码最前面加入代码
注意 ,这里为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有差距。