极客前沿

AI入门:5分钟装好环境,跑通你的第一个模型

2026-05-12 01:36
DEV Machine Learning
查看原文

本文手把手教你从零搭建AI环境、安装必要工具,并运行第一个示例模型,适合零基础读者。

准备环境:你需要装什么?

在开始之前,先了解几个关键工具:Python 是一种编程语言,很多AI工具都用它;pip 是Python的软件包管理器,负责安装别人写好的代码库;虚拟环境 就像一个独立的工具箱,让你在不同项目里用不同版本的软件而不冲突。

  • 访问 python.org,下载Python 3.9或更高版本(建议3.10)。安装时勾选“Add Python to PATH”。
  • 打开终端(Windows用cmd或PowerShell,Mac/Linux用Terminal),输入 python --version 确认安装成功。
  • 安装虚拟环境工具:pip install virtualenv

常见坑:Windows用户如果遇到“pip不是内部命令”,说明Python没加到PATH,重新安装并勾选那个选项即可。

安装步骤:一步步来

  1. 创建一个新文件夹,比如 ai-first,进入该文件夹。
  2. 创建虚拟环境:python -m venv venv。这会在文件夹里生成一个 venv 子文件夹。
  3. 激活虚拟环境:
    • Windows:venv\Scripts\activate
    • Mac/Linux:source venv/bin/activate
    看到终端前面出现 (venv) 就对了。
  4. 安装一个简单的AI库:pip install transformers。这个库让你轻松使用各种预训练模型。
  5. 安装完成后,测试一下:python -c "from transformers import pipeline; print('OK')"。如果输出“OK”,表示安装成功。

常见坑:如果安装很慢,可以加国内镜像源,例如 pip install transformers -i https://pypi.tuna.tsinghua.edu.cn/simple

Tutorial Image

验证是否成功:跑第一个模型

现在我们来运行一个真正的AI模型——文本生成模型,它可以根据你给的文字接着往下写。

  • 在终端里(确保虚拟环境激活),输入 python 进入Python交互模式。
  • 逐行输入以下代码:
    from transformers import pipeline
    generator = pipeline('text-generation', model='distilgpt2')
    result = generator('Hello, AI is', max_length=20)
    print(result)
  • 你会看到类似 [{'generated_text': 'Hello, AI is a new way to think about...'}] 的输出。恭喜,你跑通了第一个AI模型!

常见坑:第一次运行需要下载模型(约300MB),请确保网络畅通。如果内存不足,可以换更小的模型如 distilgpt2

下一步可以做什么

  • 试试修改 max_length 的值,让生成更长的文本。
  • 尝试其他任务,比如 情感分析(判断一句话是正面还是负面):classifier = pipeline('sentiment-analysis')
  • 如果想深入学习,可以阅读 Transformers官方文档

内容来源

DEV Machine Learning

发布时间

2026-05-12 01:36

返回 AI技术