tensorflow常见的GPU配置函数

1.如何指定GPU

import os
os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
# 当设置为0时表示使用第一块GPU，当使用多块GPU时，可以用类似"0,1"表示
os.environ["CUDA_VISIBLE_DEVICES"] = "0" 

2.设置GPU显存的使用率

# 指定每个进程使用GPU的上线
config = tf.ConfigProto()
config.gpu_options.per_process_gpu_memory_fraction = 0.60