在tensorflow.js中设置per_process_gpu_memory_fraction tfjs-node-gpu

Question

是否可以为tfjs-node-gpu设置最大分配GPU内存？默认情况下，它花费100％，而且我还没有查看有关API文档的任何信息。

谢谢

Answer 1

目前无法在GPU上设置内存限制；节点尚未提供对所使用GPU的控制，并且本身也没有tfjs-node-gpu的控制。

但是，您可以使用内存占用量来手动检查通过tf.memory分配的大小