用tensorflow-gpu,但是代码里没激活gpu环境。
修改:import tensorflow as tf config = tf.compat.v1.ConfigProto() config.gpu_options.allow_growth = True sess=tf.compat.v1.Session(config=config)
欢迎分享,转载请注明来源:内存溢出
用tensorflow-gpu,但是代码里没激活gpu环境。
修改:import tensorflow as tf config = tf.compat.v1.ConfigProto() config.gpu_options.allow_growth = True sess=tf.compat.v1.Session(config=config)
欢迎分享,转载请注明来源:内存溢出
评论列表(0条)