极客前沿

AI入门:一步步装好环境,跑通你的第一个AI程序

2026-08-03 01:30
DEV Beginners
查看原文

读完本文,你将学会从零开始安装AI开发环境,并成功运行第一个AI程序,迈出AI第一步。

你是不是也想试试AI,但一看教程就头大?别怕,这篇文章就是给你准备的。我们会一步步来,装好环境,跑通第一个AI程序。整个过程不难,跟着做就行。

准备环境:装好Python和编辑器

AI程序大多用Python(一种简单易学的编程语言,很多AI工具都支持它)来写。所以第一步,先装Python。

  • 去Python官网(python.org)下载最新版,安装时记得勾选“Add Python to PATH”(把Python加入系统路径,这样在命令行里才能直接使用)。
  • 装好后,打开命令行(Windows按Win+R输入cmd,Mac打开终端),输入python --version,如果显示版本号,就说明装好了。
  • 接着装个代码编辑器,推荐VS Code(一个免费好用的代码编辑软件),去官网下载安装即可。

安装AI库:用pip装TensorFlow

AI开发常用一个叫TensorFlow的库(一个由Google开发的工具包,用来构建和训练AI模型)。我们用pip(Python自带的包管理器,用来安装额外功能)来安装。

Tutorial Image
  1. 打开命令行,输入pip install tensorflow,然后回车。
  2. 等它下载安装,可能需要几分钟。如果网速慢,可以加-i https://pypi.tuna.tsinghua.edu.cn/simple(用国内镜像加速)。
  3. 装完后,输入python -c "import tensorflow as tf; print(tf.__version__)",如果输出版本号,就说明装成功了。

跑通第一个AI程序:识别手写数字

现在我们来写一个简单的AI程序,让它识别手写数字。这是AI的“Hello World”,很适合入门。

  1. 打开VS Code,新建一个文件,保存为test.py
  2. 复制以下代码粘贴进去:
import tensorflow as tf
mnist = tf.keras.datasets.mnist
(x_train, y_train), (x_test, y_test) = mnist.load_data()
x_train, x_test = x_train / 255.0, x_test / 255.0
model = tf.keras.models.Sequential([
  tf.keras.layers.Flatten(input_shape=(28, 28)),
  tf.keras.layers.Dense(128, activation='relu'),
  tf.keras.layers.Dropout(0.2),
  tf.keras.layers.Dense(10)
])
loss_fn = tf.keras.losses.SparseCategoricalCrossentropy(from_logits=True)
model.compile(optimizer='adam', loss=loss_fn, metrics=['accuracy'])
model.fit(x_train, y_train, epochs=5)
model.evaluate(x_test, y_test, verbose=2)
  1. 在命令行里进入该文件所在目录,输入python test.py运行。
  2. 看到类似“accuracy: 0.97”的输出,就说明你成功跑通了一个AI程序!

下一步:动手改一改

跑通之后,你可以试着改改代码,比如把epochs=5改成epochs=10(训练轮数,越多可能越准但更慢),或者加一行print(model.summary())(打印模型结构)。多试几次,你会更有感觉。

恭喜你,已经迈出了AI第一步。接下来,可以找些简单教程,学学怎么用AI做图片分类、文本生成等。记住,遇到问题多用搜索引擎,你并不孤单。

内容来源

DEV Beginners

发布时间

2026-08-03 01:30

返回 AI技术