TensorFlow默认会使用多个GPU吗?

时间:2017-06-25 07:45:39

标签: python tensorflow

当我没有使用“with tf.device(d):”分配设备时,TensorFlow只使用一个GPU,还是当一个GPU不够时它会使用所有可用的GPU?

1 个答案:

答案 0 :(得分:2)

TensorFlow documentation说:

  

如果系统中有多个GPU,默认情况下将选择ID最低的GPU。如果您想在不同的GPU上运行,则需要明确指定首选项。