tensorflow2.0报错:Function call stack:distributed_function

可以明确地说是GPU资源不足,更要声明的是,在使用各类平台提供的jupyter使用GPU时,开启多个tap页面时,由于Tensorflow占用资源机制是有多少就全部占用,那么在没有提前设置资源使用量时,只有第一个开启的才会成功启动,其余的都会造成上述报错

可以先关闭所有页面,然后重新启动,并进行如下配置

# import tensorflow as tf 
import os

os.environ['CUDA_VISIBLE_DEVICES']="0" # 指定哪块GPU训练
config=tf.compat.v1.ConfigProto() 
# 设置最大占有GPU不超过显存的80%(可选)
config.gpu_options.per_process_gpu_memory_fraction=0.5
config.gpu_options.allow_growth = True  # 设置动态分配GPU内存
sess=tf.compat.v1.Session(config=config)

你可能感兴趣的:(bug修复,keras,tensorflow)