2024-09-29 07:18:31 发布
网友
有没有办法在创建操作后分配设备?所以不要使用withtf.device(u'/gpu:1“)例如,我们可以使用op.device=u”/gpu:1'(显然这不起作用,但我想知道是否有类似的情况)
注意:op是一个tensorflow操作,其中type(op)返回tensorflow.python.framework.ops.operation
谢谢
目前没有回答
目前没有回答
相关问题 更多 >
编程相关推荐