tensorflow设置使用gpu

在tensorflow中，我们可以使用 tf.device() 指定模型运行的具体设备，可以指定运行在GPU还是CUP上，以及哪块GPU上。

 

设置使用GPU

使用 tf.device('/gpu:1') 指定Session在第二块GPU上运行

 

默认是gpu:0

posted @ 2021-10-15 14:58  Tomorrow1126  阅读(769)  评论(0)  编辑  收藏  举报