知道您可以设置最大Tensorflow的GPU的内存量 我想知道如何防止tensorflow **。js **使用我的所有GPU RAM。我没有在API文档中找到任何内容。
答案 0 :(得分:0)
还没有。当前,您只能使用tf.setBackend
选择注册tensorflow后端,它将使用所有可用的gpu。
tf.setBackend('tensorflow')
tf.setBackend('cpu')
tf.setBackend('webgl') // if using tfjs with the browser