分类：tensorflow指定CPU与GPU运算

2020
09-27

tensorflow指定CPU与GPU运算的方法实现

1.指定GPU运算如果安装的是GPU版本，在运行的过程中TensorFlow能够自动检测。如果检测到GPU，TensorFlow会尽可能的利用找到的第一个GPU来执行操作。如果机器上有超过一个可用的GPU，除了第一个之外的其他的GPU默认是不参与计算的。为了让TensorFlow使用这些GPU，必须将OP明确指派给他们执行。with......device语句能够用来指派特定的CPU或者GPU执行操作：importtensorflowastfimportnumpyasnpwithtf.Session()assess:wi... 继续阅读 >