1,节省内显存:
在tf.Session加入:
config = tf.ConfigProto()
config.gpu_options.allow_growth=True
sess = tf.Session(config=config)
2,运行时:
指定运行在某个GPU上面
CUDA_VISIBLE_DEVICES=0 python a.py
1,节省内显存:
在tf.Session加入:
config = tf.ConfigProto()
config.gpu_options.allow_growth=True
sess = tf.Session(config=config)
2,运行时:
指定运行在某个GPU上面
CUDA_VISIBLE_DEVICES=0 python a.py