tensorflow中使用tf.ConfigProto()配置Session運行參數&&GPU設備指定

1. 使用tf.ConfigProto()配置Session運行參數

• 記錄設備指派情況：tf.ConfigProto(log_device_placement=True)
• 自動選擇運行設備： tf.ConfigProto(allow_soft_placement=True)
• 限制GPU資源使用：

（1）動態申請顯存

config = tf.ConfigProto()
config.gpu_options.allow_growth = True
session = tf.Session(config=config)

（2）限制GPU的使用率

config = tf.ConfigProto()
config.gpu_options.per_process_gpu_memory_fraction = 0.4  #占用40%顯存
session = tf.Session(config=config)

2. GPU的設備指定

• 設置使用哪塊GPU
  （1）在程序中設置
  （2）運行時設置

CUDA_VISIBLE_DEVICES=0,1 python yourcode.py

3. 參考

https://blog.csdn.net/dcrmg/article/details/79091941