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

Question

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

标签： tensorflowtensorflow.js

Answer 1

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

• 通过使用 tensorflow 后端

tf.setBackend('tensorflow') 

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

tf.setBackend('cpu')

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