允许在张量流中进行软放置

时间:2018-01-21 16:56:09

标签: python-2.7 tensorflow

我刚刚从tf 1.1升级到tf 1.4 for python 2.7,我遇到了以下问题:

我有一个图表,我通过使用tf.device(' device')命令将所有OP放在特定设备中。但是,其中一个OP只允许在CPU设备中,所以我使用allow_soft_placement = True并且它在tf 1.1中正常工作(它只放置了没有GPU实现的OP和GPU中的其他OP)。但是现在(在tf1.4中)当我运行我的网络时,它将所有的OP放在CPU中(而不仅仅是没有GPU实现的那个)。

感谢任何帮助。

1 个答案:

答案 0 :(得分:0)

可能会发生此错误,因为您使用的是python 2.7。其中tensorflow用于python 3.5和python 3.6