经验:tensorflow 的 GPU 配置

由于tensorflow-gpu会默认占据所有的GPU显存,所以,在编写代码时,最好添加以下代码,使得占据的显存按需增长,而不是不论需要多少都占据所有显存

os.environ["CUDA_VISIBLE_DEVICES"] = '0'
config = tf.ConfigProto()
config.gpu_options.allow_growth=True
sess = tf.Session(config=config)

 

猜你喜欢

转载自blog.csdn.net/Strive_For_Future/article/details/81295551