极客前沿

AI 入门:第一次装环境、跑模型,手把手教程

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

读完本文,你将从零开始安装 Python、配置虚拟环境,并成功运行一个简单的开源语言模型。

准备环境:装 Python 和 Git

在跑任何 AI 模型之前,我们需要先装好两个基础工具:Python(一种编程语言,AI 模型大多用它写)和 Git(一个下载代码的工具)。

  • 下载 Python:去 python.org,点击“Download”,选最新版本(比如 3.10 或更高)。安装时记得勾选“Add Python to PATH”。
  • 下载 Git:去 git-scm.com,下载对应系统版本,一路默认安装。
  • 验证安装:打开终端(Windows 按 Win+R 输入 cmd,Mac 打开“终端”),输入 python --versiongit --version,能看到版本号就对了。

安装步骤:下载模型代码并装依赖

我们用 Hugging Face 上的一个轻量模型做演示,它叫 distilgpt2(一个简化版的语言模型,能生成文本)。

Tutorial Image
  1. 打开终端,输入 pip install transformers torch,按回车。这会安装两个核心库:transformers(让模型跑起来的工具包)和 torch(计算引擎)。
  2. 创建一个新文件夹:mkdir my_first_ai 然后 cd my_first_ai
  3. 新建一个 Python 文件,比如 run_model.py,用记事本打开,复制下面代码:
    from transformers import pipeline
    
    generator = pipeline('text-generation', model='distilgpt2')
    result = generator("Hello, I'm learning AI", max_length=50)
    print(result[0]['generated_text'])
  4. 保存后,在终端输入 python run_model.py,等待几秒下载模型(第一次会慢些),然后你会看到模型生成的续写文字。

验证是否成功:看输出

如果终端打印出了一段英文文本(比如“Hello, I'm learning AI and it's fun...”),恭喜你,环境跑通了!常见坑:

  • 如果报错“No module named 'transformers'”,说明没装成功,重新运行 pip install transformers
  • 如果下载模型超时,可以换个网络,或使用国内镜像:在 pip 命令后加 -i https://pypi.tuna.tsinghua.edu.cn/simple

下一步可以做什么

你已经迈出了第一步!接下来可以:

  • 换一个模型:在 model='distilgpt2' 里改成 'gpt2''bert-base-uncased',看看不同效果。
  • 学习微调:用你自己的文本数据训练模型,让它学会写特定风格的内容。
  • 了解模型来源:就像新闻里提到的,很多模型是站在巨人肩膀上开发的,用之前看看文档,尊重原创。

内容来源

DEV Machine Learning

发布时间

2026-06-15 01:32

返回 AI技术