TensorFlow:Google 开源的机器学习平台

TensorFlow 在 GitHub 上拿到 195,604 Star。

这个项目做了一件事:给研究者和开发者提供一套完整的机器学习工具链,从训练到部署,全链路覆盖。

1、 背景

TensorFlow 最早是 Google Brain 团队内部用的,做神经网络研究。后来发现框架通用性够强,直接开源了。

现在它覆盖的场景远超最初的目标。计算机视觉、自然语言处理、推荐系统、强化学习,各个方向都有团队在用。

正文顶部截图

2、 核心能力

TensorFlow 提供稳定的 Python 和 C++ API,其他语言也有非保证向后兼容的接口。

整个生态分三块:工具层有 TensorBoard 做可视化、TFX 做生产流水线;库和扩展层有 TF Lite 跑移动端、TF.js 跑浏览器;社区层有教程、预训练模型、Codelabs。

GPU 支持是标配,NVIDIA CUDA 显卡直接加速,macOS Metal 和 DirectX 通过插件也能跑。

3、 安装

最简单的方式:

pip install tensorflow

GPU 支持需要系统装好对应的 CUDA 驱动。只用 CPU 可以装精简包:

pip install tensorflow-cpu

不想折腾环境,Docker 镜像现成的。想从源码编译,官方文档有完整步骤。

4、 一个例子

import tensorflow as tf
tf.add(1, 2).numpy()   # 输出: 3
hello = tf.constant('Hello, TensorFlow!')
hello.numpy()           # 输出: b'Hello, TensorFlow!'

几行代码跑通。实际项目当然复杂得多,但入门门槛确实不高。

README区域截图

5、 适合谁用

做机器学习研究的人,需要一个成熟的训练框架。做工程部署的人,需要从实验到生产的完整路径。在企业里搭 AI 系统的人,需要社区活跃、文档齐全、坑少的方案。TensorFlow 在这几个方向上都能撑住。

Apache License 2.0,商用无限制。

的方案。TensorFlow 在这几个方向上都能撑住。

Apache License 2.0,商用无限制。

更多推荐