Tensorflow:如何为冻结图形中的操作分配设备?

2024-06-16 11:16:34 发布

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

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

有人能为
提供一点python代码的和平吗 1) 按名称查找操作
2) 将其分配给CPU设备

谢谢你的预付款


Tags: 文件代码名称图形gpucpupb预付款