python - Keras Tensorflow Graph 在 GPU 和 CPU 之间拆分
问题描述
我很好奇 Keras 如何在后台处理 CPU 和 GPU 分配。不幸的是,Keras 网站上的文档在这方面非常有限。
我的Embedding
层在内存中拟合时遇到问题,所以我按照这个答案中的说明,只强制我的Embedding
层在 CPU 上,其余的(密集层等)在 GPU 上。我现在没有任何问题,但很好奇这是如何工作的。
CPU 和 GPU 上是否有图表版本?还是在嵌入完成后将数据传递到 GPU 上的其余图形?如果是这样,如何训练重量?
解决方案
推荐阅读
- apache-spark - 当我尝试根据条件修改列时出现 Pyspark 错误,否则
- java - 我的 IDE 中是否存在错误 - 无法使用 Neatbeans IDE 在控制台中输入值
- node.js - 部署到 Heroku 时,在 typescript express 和 react 应用程序中找不到 tsc
- cmake - 在项目名称后设置时 CMAKE_CXX_STANDARD 不起作用
- python - 更改对数轴上的刻度线大小
- postgresql - Postgresql 计算记录之间的最小时间差
- azure - 如何禁用 Azure BLOB 静态网站上的目录列表?
- vim - 在 VIM 中使用插入时出现延迟
- node.js - GSI 上的 DynamoDB 查询使用 nodeJs 返回 null
- python - 用于测试 setUp 中所有实例的 Python unittest 在属性中是否具有唯一编号?