TensorFlow.js 是一个 JavaScript 库,用于在浏览器和 Node.js 训练和部署机器学习模型。
浏览器设置安装 TensorFlow.js
您可以通过两种主要方式在浏览器项目中获取 TensorFlow.js:
使用脚本代码
将以下脚本代码添加到您的主 HTML 文件中。
<script src="https://cdn.jsdelivr.net/npm/@tensorflow/tfjs@2.0.0/dist/tf.min.js"></script>
从 NPM 安装
您可以使用 npm cli 工具或 yarn 安装 TensorFlow.js。
yarn add @tensorflow/tfjs
或
npm install @tensorflow/tfjs
Node.js 设置安装 TensorFlow.js
您可以使用 npm cli 工具或 yarn 安装 TensorFlow.js。
选项 1:使用原生 C++ 绑定安装 TensorFlow.js。
yarn add @tensorflow/tfjs-node
或
npm install @tensorflow/tfjs-node
选项 2:(仅限 Linux)如果您的系统搭载具备 CUDA 支持的 NVIDIA® GPU,请使用 GPU 软件包以获得更高性能。
yarn add @tensorflow/tfjs-node-gpu
或
npm install @tensorflow/tfjs-node-gpu
选项 3:安装纯净版 JavaScript。这是性能最慢的选项。
yarn add @tensorflow/tfjs
或
npm install @tensorflow/tfjs
TypeScript
使用 TypeScript 时,如果您的项目使用严格的 null 检查,则可能需要在 tsconfig.json 文件中设置 skipLibCheck: true,否则在编译过程中会遇到错误。