极客前沿

AI入门:从零开始搭建你的第一个模型

2026-06-26 01:30
DEV Beginners
查看原文

本文为零基础读者介绍AI环境安装、第一个模型训练与运行的全过程,包含步骤和常见坑。

很多新手想学AI,但第一步就被环境安装劝退了。其实,只要跟着下面几个步骤,你也能在半小时内跑通第一个AI程序。

Tutorial Image

准备环境

首先,你需要一台电脑(Windows/Mac/Linux都可以),然后安装 Python(一种编程语言,AI界最常用)。下载地址:python.org,建议选3.8以上版本。安装时记得勾选“Add Python to PATH”。

安装步骤

  1. 打开终端(Windows按Win+R,输入cmd;Mac打开“终端”)。
  2. 输入 python --version,确认版本号显示正确。
  3. 安装 pip(Python的包管理工具,帮你下载别人写好的代码库)。通常Python自带pip,输入 pip --version 检查。
  4. 安装 TensorFlow(一个AI框架,帮你快速搭建神经网络):pip install tensorflow。等待下载完成(约几分钟)。
  5. 安装 Jupyter Notebook(一个交互式编程环境,方便写和运行代码):pip install jupyter

运行第一个AI程序

  1. 终端输入 jupyter notebook,浏览器会自动打开一个页面。
  2. 点击右上角“New” -> “Python 3”,新建一个笔记本。
  3. 复制以下代码到第一个单元格:
import tensorflow as tf
print("TensorFlow版本:", tf.__version__)
# 创建一个简单的模型
model = tf.keras.Sequential([tf.keras.layers.Dense(10, activation='relu', input_shape=(10,))])
print("模型创建成功!")
  1. 按Shift+Enter运行,看到输出即可。

常见坑提醒

  • 网络问题:如果pip下载慢,可以换国内镜像:pip install -i https://pypi.tuna.tsinghua.edu.cn/simple tensorflow
  • 版本冲突:如果报错,尝试升级pip:pip install --upgrade pip
  • 显卡驱动:如果你有NVIDIA显卡,可以额外安装CUDA和cuDNN加速训练,但新手可以先跳过。

现在你已经成功跑通了AI环境!下一步可以学习如何训练一个图像分类模型,比如识别手写数字。

内容来源

DEV Beginners

发布时间

2026-06-26 01:30

返回 AI技术