AI 入门:一步步安装 TensorFlow 并跑通第一个模型
零基础学会安装 Python 和 TensorFlow,运行一个简单的图像分类模型,体验 AI 实战。
想学 AI 但不知道从哪开始?别担心,这篇教程会带你一步步装好环境、跑通第一个模型。我们用的是 Google 开源的 TensorFlow(一个帮电脑学习识别图片、文字等数据的工具库)。读完本文,你就能在自己的电脑上运行一个简单的图像分类程序了。
1. 准备环境:安装 Python
TensorFlow 需要 Python(一种简单易学的编程语言)来运行。请按以下步骤操作:
- 打开浏览器,访问 python.org/downloads,下载 Python 3.8 或更高版本(推荐 3.9 或 3.10)。
- 运行安装包,务必勾选“Add Python to PATH”(这样电脑才知道 Python 命令在哪)。
- 安装完成后,打开命令提示符(Windows 按 Win+R,输入 cmd 回车)或终端(Mac/Linux),输入
python --version,看到版本号就说明装好了。
2. 安装 TensorFlow
现在用 Python 自带的 pip(一个安装软件包的工具)来装 TensorFlow。
- 在命令提示符或终端里输入:
pip install tensorflow,然后回车。 - 等待下载完成(可能需要几分钟,网速慢的话请耐心)。
- 安装成功后,输入
pip show tensorflow,会显示版本号等信息。
常见坑:如果报错“pip 不是内部或外部命令”,说明 Python 没加到 PATH,请重新安装并勾选那个选项。
3. 跑通第一个模型:识别手写数字
我们用一个经典例子来验证安装是否成功——用 TensorFlow 识别手写数字图片。
- 新建一个文本文件,命名为
test.py。 - 把下面的代码复制进去:
import tensorflow as tf
(train_images, train_labels), (test_images, test_labels) = tf.keras.datasets.mnist.load_data()
model = tf.keras.models.Sequential([
tf.keras.layers.Flatten(input_shape=(28, 28)),
tf.keras.layers.Dense(128, activation='relu'),
tf.keras.layers.Dense(10)
])
model.compile(optimizer='adam', loss='sparse_categorical_crossentropy', metrics=['accuracy'])
model.fit(train_images, train_labels, epochs=5)
test_loss, test_acc = model.evaluate(test_images, test_labels)
print('测试准确率:', test_acc) - 在命令提示符或终端里,进入文件所在目录,输入
python test.py回车。 - 看到一堆数字滚动,最后打印出“测试准确率: 0.97”左右,就说明跑通了!
解释一下:这段代码让电脑看几千张手写数字图片(0-9),并学会识别它们。准确率 97% 意味着电脑几乎不会认错。
4. 下一步可以做什么
- 试试用你自己的图片:找一张数字图片,用
model.predict()让它猜猜看。 - 学习更多 TensorFlow 教程,比如识别猫狗照片。
- 别忘了关掉命令提示符,不然电脑会一直跑程序哦。
恭喜你,已经迈出了 AI 入门的第一步!
内容来源
DEV Tutorial
发布时间
2026-07-15 01:46