tf.device()指定tensorflow运行的GPU或CPU设备
转自:牧野
在tensorflow中,我们可以使用 tf.device() 指定模型运行的具体设备,可以指定运行在GPU还是CUP上,以及哪块GPU上。设置使用GPU使用 tf.device('/gpu:1') 指定Session在第二块GPU上运行:import tensorflow as tf with tf.device('/gpu:1'): v1 =...
转载
2019-04-11 15:38:36 ·
1291 阅读 ·
1 评论