Node.js中使用TensorFlow.js报No backend found in registry错误求助

这个错误的核心原因是：Node.js环境中，TensorFlow.js不会默认内置计算后端，必须手动安装并注册对应后端包。浏览器用的CDN版本依赖WebGL后端，在Node.js环境无法生效，所以你用CDN也会报错。

解决步骤： #

1. 安装Node.js专属后端包

   • 若使用CPU运行：

     npm install @tensorflow/tfjs @tensorflow/tfjs-node
     

   • 若有NVIDIA GPU且已配置CUDA环境（可加速运算）：

     npm install @tensorflow/tfjs @tensorflow/tfjs-node-gpu
     

   注意：确保@tensorflow/tfjs和tfjs-node/tfjs-node-gpu的版本完全一致，否则可能出现兼容性问题。

2. 在代码中正确导入后端
   直接导入@tensorflow/tfjs-node或tfjs-node-gpu即可，它会自动注册后端，无需额外手动注册：

   const tf = require('@tensorflow/tfjs-node');
   const mobilenet = require('@tensorflow-models/mobilenet');
   const fs = require('fs');
   
   async function classifyImage(imagePath) {
     // 读取本地图像并转为张量
     const imageBuffer = fs.readFileSync(imagePath);
     const inputTensor = tf.node.decodeJpeg(imageBuffer, 3);
   
     // 加载MobileNet模型
     const model = await mobilenet.load();
     // 执行分类
     const predictions = await model.classify(inputTensor);
   
     console.log('分类结果:', predictions);
     // 释放张量内存
     inputTensor.dispose();
   }
   
   // 调用分类函数，替换为你的图像路径
   classifyImage('./your-image.jpg');
   

3. 验证后端是否生效
   可以在代码开头添加以下代码，确认后端已注册：

   console.log('可用后端:', tf.getBackend());
   

   正常情况下会输出node（CPU版）或node-gpu（GPU版）。

内容的提问来源于stack exchange，提问作者Baya Osborn