有没有办法设置per_process_gpu_memory_fraction tensorflow.js（.JS！）？

Question

知道您可以设置最大Tensorflow的GPU的内存量 我想知道如何防止tensorflow **。js **使用我的所有GPU RAM。我没有在API文档中找到任何内容。

Answer 1

还没有。当前，您只能使用tf.setBackend选择注册tensorflow后端，它将使用所有可用的gpu。

• 通过使用tensorflow后端

tf.setBackend('tensorflow') 

通过使用
• 在浏览器中使用webgl或cpu

tf.setBackend('cpu')

tf.setBackend('webgl') // if using tfjs with the browser