极客前沿

AI入门:一步步装好深度学习环境,跑通你的第一个模型

2026-06-26 01:32
DEV Machine Learning
查看原文

手把手教零基础读者安装Python、TensorFlow,并运行第一个AI示例代码。

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

开始学AI,首先要装好Python(一种编程语言,AI最常用的工具之一)。建议去官网 python.org 下载最新版,安装时记得勾选“Add Python to PATH”。装完后打开命令行,输入 python --version 检查是否成功。接着装个编辑器,推荐VS Code(免费好用的代码编辑器),去 code.visualstudio.com 下载安装即可。

安装核心库:TensorFlow和Jupyter

打开命令行(Windows搜索cmd,Mac/Linux打开终端),输入以下命令安装TensorFlow(一个帮你搭建和训练AI模型的工具包):pip install tensorflow。再装Jupyter Notebook(一种交互式编程环境,方便边写代码边看结果):pip install jupyter。安装过程可能有点慢,耐心等待,如果报错多半是网络问题,可以换国内镜像源,比如 pip install tensorflow -i https://pypi.tuna.tsinghua.edu.cn/simple

跑通第一个AI程序

在命令行输入 jupyter notebook,浏览器会自动打开一个页面。点击右上角“New” -> “Python 3”新建一个笔记本。复制下面这段代码进去,然后按Shift+Enter运行:

Tutorial Image
import tensorflow as tf
print(tf.__version__)

如果输出了类似 2.x.x 的数字,说明环境装好了!再试试一个真正的AI示例——用神经网络(一种模仿人脑的算法)识别手写数字:

mnist = tf.keras.datasets.mnist
(x_train, y_train), (x_test, y_test) = 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, activation='softmax')
])
model.compile(optimizer='adam', loss='sparse_categorical_crossentropy', metrics=['accuracy'])
model.fit(x_train, y_train, epochs=3)
print('训练完成!')

这段代码会下载手写数字数据集,训练一个简单模型。看到“训练完成!”就成功了。

下一步可以做什么

  • 学更多AI概念:推荐搜索“TensorFlow 官方教程中文版”。
  • 尝试改代码:比如把 epochs=3 改成 epochs=5(训练轮数),看准确率变化。
  • 保存模型:用 model.save('my_model.h5') 把训练好的模型存下来。

你已经迈出了AI第一步,接下来就是多动手、多尝试!

内容来源

DEV Machine Learning

发布时间

2026-06-26 01:32

返回 AI技术