标签: python tensorflow
我有一个简单的图形,其中包含2个Conv2D操作。 该图将冻结在.pb文件中。 加载后,两个Conv2D操作都分配给GPU:0 我想将第一个分配给CPU。
也许有人可以为python代码提供些许安全感 1)按名称查找操作 2)将其分配给CPU设备
感谢前进!