📜  Tensorflow.js 简介

📅  最后修改于: 2022-05-13 01:56:42.308000             🧑  作者: Mango

Tensorflow.js 简介

什么是 Tensorflow.js?

TensorFlow.js 是一个 JavaScript 库,用于在 Web 应用程序和 Node.js 中训练和部署机器学习模型。您可以使用 tensorflow.js 从头开始开发机器学习模型,也可以使用提供的 API 在浏览器或 Node.js 服务器上训练现有模型。

要了解更多信息,您可以直接访问此链接:https://www.tensorflow.org/resources/learn-ml/basics-of-tensorflow-for-js-development。

先决条件:在启动 Tensorflow.js 之前,您需要了解以下内容:

对于浏览器:

  • HTML:需要 HTML 的基础知识
  • JavaScript:需要良好的 JS 知识

对于服务器端:

  • Node.js:熟练掌握 Node.js。此外,由于 Node.js 是一个 JS 运行时,因此对 JavaScript 进行命令会很有帮助。

其他需求:

  • NPM 或 Yarn:这些是需要安装在系统中的软件包。

设置 Tensorflow.js:

浏览器设置在基于浏览器的应用程序中添加 TensorFlow.js 有两种方法:

  • 使用脚本标签。
  • 从 NPM 安装

1. 使用脚本标签:将以下脚本标签添加到您的主 HTML 文件中。

例子:

HTML


  

    

  

    

  



输出:

2. 使用 NPM/yarn:我们可以使用 NPM 或 Yarn 来安装 tensorflow.js。

yarn add @tensorflow/tfjs

要么

npm install @tensorflow/tfjs

Node.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