如何调用tensorflow的gpu版本
发布网友
发布时间：2022-04-21 22:43
共1个回答
热心网友
时间：2023-09-18 02:01
Tensor-flow 默认使用一个gpu进行运算（如果有gpu）。
具体的调用会在程序里。
如果要使用多gpu要相应修改程序。
至于哪部份会放到gpu上跑，一般来说是model training的部分，因为这部分程序是computation intense。
可以把其他部分放到gpu上运行，但是由于cpu-gpu的通信时间较长，会使得程序执行的更慢。
