由于tensorflow-gpu会默认占据所有的GPU显存,所以,在编写代码时,最好添加以下代码,使得占据的显存按需增长,而不是不论需要多少都占据所有显存
os.environ["CUDA_VISIBLE_DEVICES"] = '0'
config = tf.ConfigProto()
config.gpu_options.allow_growth=True
sess = tf.Session(config=config)
由于tensorflow-gpu会默认占据所有的GPU显存,所以,在编写代码时,最好添加以下代码,使得占据的显存按需增长,而不是不论需要多少都占据所有显存
os.environ["CUDA_VISIBLE_DEVICES"] = '0'
config = tf.ConfigProto()
config.gpu_options.allow_growth=True
sess = tf.Session(config=config)