有没有办法设置 per_process_gpu_memory_fraction tensorflow.js (.JS!)?
Is there a way to set per_process_gpu_memory_fraction tensorflow.js (.JS!)?
知道你可以设置最大值。 Tensorflow 的 GPU 内存量,
我想知道如何防止 tensorflow**.js** 使用我所有的 gpu ram。我在 API 文档中没有找到任何内容。
还没有。目前,您只能 select 使用 tf.setBackend
注册 tensorflow 后端,这将使用所有可用的 gpu。
- 使用tensorflow后端
tf.setBackend('tensorflow')
- 在浏览器中使用 webgl 或 cpu
tf.setBackend('cpu')
tf.setBackend('webgl') // if using tfjs with the browser
知道你可以设置最大值。 Tensorflow 的 GPU 内存量, 我想知道如何防止 tensorflow**.js** 使用我所有的 gpu ram。我在 API 文档中没有找到任何内容。
还没有。目前,您只能 select 使用 tf.setBackend
注册 tensorflow 后端,这将使用所有可用的 gpu。
- 使用tensorflow后端
tf.setBackend('tensorflow')
- 在浏览器中使用 webgl 或 cpu
tf.setBackend('cpu')
tf.setBackend('webgl') // if using tfjs with the browser