有没有办法设置per_process_gpu_memory_fraction tensorflow.js(.JS!)?

时间:2019-02-13 22:48:50

标签: tensorflow tensorflow.js

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

1 个答案:

答案 0 :(得分:0)

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

  • 通过使用tensorflow后端
tf.setBackend('tensorflow') 
    通过使用
  • 在浏览器中使用webgl或cpu
tf.setBackend('cpu')

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