TensorFlow.js WebGPU后端报错：非localhost部署时找不到后端名称

核心原因 #

WebGPU API有严格的**安全上下文（Secure Context）**要求：仅允许在HTTPS协议环境或浏览器默认信任的localhost/127.0.0.1中运行。当改用本地IP（如192.168.x.x）通过HTTP部署时，页面处于非安全上下文，tfjs-backend-webgpu会因为环境不满足要求自动跳过注册流程，导致调用tf.setBackend("webgpu")时出现“后端未找到”的错误。

解决方案 #

1. 改用HTTPS部署本地服务器（推荐） #

给本地服务器配置HTTPS证书（可生成自签名证书用于开发），就能让本地IP的连接进入安全上下文：

• 使用http-server：

  http-server -S -C 你的证书文件.pem -K 你的私钥文件.key
  

• 使用Express：

  const https = require('https');
  const fs = require('fs');
  const express = require('express');
  const app = express();
  
  const options = {
    key: fs.readFileSync('私钥文件.key'),
    cert: fs.readFileSync('证书文件.pem')
  };
  
  https.createServer(options, app).listen(443);
  

2. 临时修改浏览器配置（仅开发测试用） #

如果只是临时测试，可以通过浏览器启动参数强制允许非安全上下文使用WebGPU：

• Chrome/Edge启动时添加参数：

  --enable-features=WebGPU,AllowWebGPUOnInsecureOrigins
  

  注意：此方法仅适合本地开发调试，生产环境必须使用HTTPS。

常见疑问解答 #

• 是否需要特定HTTP头？不需要额外配置HTTP头，核心是满足安全上下文要求即可。
• 为什么仅修改IP会引发问题？因为localhost/127.0.0.1是浏览器默认豁免的安全上下文，而普通HTTP协议的本地IP连接不属于安全范畴，触发了WebGPU的环境限制。

你的测试代码 #

await import(`https://cdn.jsdelivr.net/npm/@tensorflow/tfjs@4.17.0/dist/tf.min.js`);
await import(`https://cdn.jsdelivr.net/npm/@tensorflow/tfjs-backend-webgpu/dist/tf-backend-webgpu.js`);
await tf.ready();
await tf.setBackend("webgpu");   // <--Error: Backend name 'webgpu' not found in registry
console.log(tf.getBackend());

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