【发布时间】：2019-12-19 23:30:51
【问题描述】：
是否可以为 tfjs-node-gpu 设置最大分配 GPU 内存？默认是 100%，我没有查看 API 文档的任何信息。
谢谢
【问题讨论】：
标签： tensorflow.js
【发布时间】：2019-12-19 23:30:51
【问题描述】：
目前无法在 GPU 上设置内存限制； node 还没有提供对所使用的 gpu 的控制，tfjs-node-gpu 本身也没有。
但是，您可以使用内存占用来手动检查tf.memory分配的大小
【讨论】：