TensorFlow.js 可以在瀏覽器或 NodeJS 上 training 和 deploy models。

如果不想涉獵低階的 Tensors 或 Optimizers,可以先從高階的ml5.js 開始, 這個 ml5.js 是 open source 並封裝了 TensorFlow.js 不需要安裝其他 dependencies,可以直接使用內建的 models & algorithms。

TensorFlow JS 安裝方式

% npm install @tensorflow/tfjs-node     # for CPU
% npm install @tensorflow/tfjs-node-gpu # for GPU
% npm install @tensorflow/tfjs          # 效能最糟
import * as tf from '@tensorflow/tfjs-core';
import '@tensorflow/tfjs-backend-webgl';

如果在 Node.js 中,如果有支援 CUDA (EX: NVIDIA GPU) 或者在 Browser 中有 WebGL, TensorFlow.js 是支援硬體加速的。以WebGL 來說 GPU 預設已經包含在 @tensorflow/tfjs, 只要呼叫以下 setBackend()

    tf.setBackend('webgl');

今日安裝的 TensorFlow.js v3.18,如果因為型態錯誤而編譯失敗 必須要在 tsconfig.json 新增 “skipLibCheck”: true 避免 Long 及相關計算形態上的錯誤。

可以好好開始來嚐鮮一下囉 ~