import tensorflow  as tf

  #查看显卡编号映射

sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))

相关文章: