如何确认TensorFlow.js是否启用SIMD？Chrome 107环境检测方法

一、检查浏览器原生SIMD支持 #

直接在Chrome开发者工具的Console面板中执行以下代码：

WebAssembly.supportedFeatures().has('simd')

返回true则表示当前Chrome 107支持WebAssembly SIMD特性。

二、确认TensorFlow.js是否启用SIMD加速 #

TensorFlow.js会自动适配环境启用SIMD（若支持），可通过两种方式验证：

• 方式1：查看后端特性
  在Console中依次执行：

  tf.getBackend()
  

  若返回wasm（Movenet默认使用WASM后端），继续执行：

  tf.backend().features
  

  查看返回对象中是否存在SIMD: true的属性，存在则说明TFJS已启用SIMD加速。
• 方式2：查看启动日志
  在代码最开头添加日志级别设置：

  tf.setLogLevel('info')
  

  运行Movenet模型后，在Console中查看是否有Using SIMD WASM backend类的日志输出，出现则表示SIMD已启用。

三、验证Movenet是否实际利用SIMD加速 #

通过性能对比测试确认：

1. 先记录启用SIMD时的推理耗时：用performance.now()计算单帧姿态估计的时间，比如：

   const start = performance.now();
   await detector.estimatePoses(video);
   const end = performance.now();
   console.log('启用SIMD耗时:', end - start);
   

2. 临时禁用SIMD（仅测试用）：在加载TensorFlow.js之前添加以下代码：

   window.tf = { setWasmPath: () => {}, wasm: { SIMD_DISABLE: true } };
   

3. 再次运行模型并记录耗时，若启用SIMD后的耗时明显低于禁用状态，说明Movenet确实在使用SIMD加速。

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