AI入门:5分钟装好环境,跑通你的第一个模型
本文手把手教你从零搭建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,重新安装并勾选那个选项即可。
安装步骤:一步步来
- 创建一个新文件夹,比如
ai-first,进入该文件夹。 - 创建虚拟环境:
python -m venv venv。这会在文件夹里生成一个venv子文件夹。 - 激活虚拟环境:
- Windows:
venv\Scripts\activate - Mac/Linux:
source venv/bin/activate
(venv)就对了。 - Windows:
- 安装一个简单的AI库:
pip install transformers。这个库让你轻松使用各种预训练模型。 - 安装完成后,测试一下:
python -c "from transformers import pipeline; print('OK')"。如果输出“OK”,表示安装成功。
常见坑:如果安装很慢,可以加国内镜像源,例如 pip install transformers -i https://pypi.tuna.tsinghua.edu.cn/simple。
验证是否成功:跑第一个模型
现在我们来运行一个真正的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