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