Tensorflow:如何在冻结图中分配用于操作的设备?

时间:2019-11-24 17:30:05

标签: python tensorflow

我有一个简单的图形,其中包含2个Conv2D操作。
该图将冻结在.pb文件中。
加载后,两个Conv2D操作都分配给GPU:0
我想将第一个分配给CPU。

也许有人可以为python代码提供些许安全感
1)按名称查找操作
2)将其分配给CPU设备

感谢前进!

0 个答案:

没有答案