TensorFlow:Google 开源的机器学习平台
·
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!'
几行代码跑通。实际项目当然复杂得多,但入门门槛确实不高。

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


所有评论(0)