通过属性将设备分配给Tensorflow中的操作

2024-09-29 07:18:31 发布

您现在位置:Python中文网/ 问答频道 /正文

有没有办法在创建操作后分配设备?所以不要使用withtf.device(u'/gpu:1“)例如,我们可以使用op.device=u”/gpu:1'(显然这不起作用,但我想知道是否有类似的情况)

注意:op是一个tensorflow操作,其中type(op)返回tensorflow.python.framework.ops.operation

谢谢


Tags: gpudevicetensorflowtype情况frameworkoperationops