AI入门:从零安装到跑通第一个模型
手把手教你安装Python和Transformers库,并运行你的第一个AI模型。
准备工作:你需要什么
在开始之前,确保你的电脑能联网,并且有至少4GB内存。我们将使用Python(一种编程语言,就像电脑的通用语言)来运行AI模型。如果你没有Python,别担心,下面会教你安装。
第一步:安装Python
- 打开浏览器,访问 python.org,点击“Downloads”下载最新版Python(建议3.9或更高)。
- 运行安装包,务必勾选“Add Python to PATH”(这样电脑才能找到Python),然后点击“Install Now”。
- 安装完成后,打开命令提示符(Windows按Win+R,输入cmd回车;Mac打开终端),输入
python --version,如果显示版本号,说明安装成功。
第二步:安装Transformers库
Transformers是一个工具包,让AI模型更容易使用。我们通过pip(Python的安装助手)来安装。
- 在命令提示符中,输入
pip install transformers,按回车。等待下载完成,中间可能会提示“Successfully installed”。 - 如果遇到“pip不是内部命令”,说明Python没加到PATH,重新安装Python并勾选那个选项。
第三步:运行你的第一个AI模型
我们将使用一个文本生成模型(能根据你给的开头自动写句子)。
- 创建一个新文件,命名为
test.py,用记事本打开。 - 复制以下代码:
from transformers import pipeline generator = pipeline('text-generation', model='gpt2') result = generator('Hello, I am learning AI', max_length=30) print(result) - 保存文件,然后在命令提示符中,进入文件所在文件夹(例如
cd Desktop),输入python test.py运行。 - 第一次运行会下载模型(约500MB),耐心等待。成功后,你会看到类似
[{'generated_text': 'Hello, I am learning AI and ...'}]的输出。
常见坑和下一步
- 下载慢:可以设置镜像源,例如
pip install transformers -i https://pypi.tuna.tsinghua.edu.cn/simple。 - 内存不足:换小模型,如
distilgpt2。 - 下一步:尝试修改输入文本,或试试其他任务(如情感分析),访问 huggingface.co/models 探索更多模型。
内容来源
DEV Machine Learning
发布时间
2026-06-29 01:31