2019年1月14日
13 tensorflow
摘要： 在构建计算图的时候，规定好在哪个设备上跑 #create a graph. with tf.device('/gpu:2'): a = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], name = 'a' ) b = tf.constant([1.0, 2.0 阅读全文
posted @ 2019-01-14 14:45 loooffeeeey 阅读(103) 评论(0) 推荐(0)