import tensorflow as tf
#查看显卡编号映射
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))