Tensorflow for Node GPU doesn't release memory

前端 未结 0 1335
无人共我
无人共我 2021-01-28 02:28

Using @tensorflow/tfjs-node-gpu 3.0.0 and CUDA 10.0 on Ubuntu 20.04

// Prepares and returns 4GB data
async function prepareData () {
  return {
    inputs: inputT         


        
相关标签:
回答
  • 消灭零回复
提交回复
热议问题