tensorflow在训练模型的时候如何指定GPU进行训练
在用tensorflow训练深度学习模型的时候,假设我们在训练之前没有指定GPU来进行训练,则默认的是选用第0块GPU来训练我们的模型,而且其它几块GPU的也会显示被占用。有些时候,我们更希望可以通过自己指定一块或者几块GPU来训练我们的模型,而不是用这种默认的方法。接下来将简单介绍两种简单的方法。
我们现有的GPU个数如下所示:
1.第一种通过tf.device()函数来指定训练时所
原创
2017-07-26 21:20:32 ·
62395 阅读 ·
10 评论