第2章 开发环境基石:Python环境搭建(Anaconda)、版本管理工具Git、代码编辑器VS Code配置
说实话,很多初学者在AIGC这条路上栽跟头,不是因为学不会模型原理,而是卡在了环境配置上。我见过太多人花了两天装环境,最后心态崩了。所以这一章,咱们就把地基打牢。
我会带你搞定三样东西:Python环境(用Anaconda)、版本管理(Git)、代码编辑器(VS Code)。这三样配齐了,后面30章你只管写代码,不用再折腾环境。
2.1 为什么非要用Anaconda?
Python本身很简单,但AIGC项目依赖的库太多了。你想想看,一个项目要PyTorch,另一个要TensorFlow,版本还互相打架。这时候就需要虚拟环境来隔离。
Anaconda就是干这个的。它自带Python,还带了一个叫conda的包管理器。我个人习惯用conda创建虚拟环境,每个项目一个环境,互不干扰。
核心概念:虚拟环境 = 独立的Python小世界。你在里面装什么库,都不会影响到外面的系统Python。
2.2 安装Anaconda
去官网下载Anaconda安装包,选Python 3.9或3.10版本。我建议别追最新版,3.11有些库还没适配好。
安装时注意两点:
- Windows用户记得勾选「Add Anaconda to my PATH environment variable」——虽然安装程序会警告,但勾上省事。
- 安装路径不要有中文和空格。我曾经因为路径带了个「项目」两个字,折腾了半天。
装完后打开终端(Windows用Anaconda Prompt,Mac/Linux用自带终端),输入:
conda --version
能看到版本号,就说明装好了。
2.3 创建你的第一个虚拟环境
咱们这门课的所有项目,我都会用同一个环境名:aigc_env。执行:
conda create -n aigc_env python=3.10
它会下载Python 3.10和一些基础包。等几分钟就好。
激活环境:
conda activate aigc_env
看到命令行前面多了 (aigc_env) 就对了。以后每次写代码,先激活这个环境。
小技巧:如果你用VS Code,可以直接在右下角选择Python解释器为 aigc_env 里的Python,不用每次都手动激活。
2.4 安装核心库
环境激活后,咱们把AIGC开发最常用的库装上:
pip install numpy pandas matplotlib jupyter
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
pip install transformers datasets accelerate
这里我装的是CPU版的PyTorch。如果你有NVIDIA显卡,把 cpu 换成 cu118 或 cu121(对应CUDA版本)。
验证一下:
python -c "import torch; print(torch.__version__)"
不报错就说明环境配好了。
2.5 Git:版本管理的救命稻草
写代码最怕什么?改了一堆东西,发现跑不通,想回退却回不去。Git就是干这个的。
我刚开始做项目时也不爱用Git,觉得麻烦。直到有一次我删了一个关键文件,找不回来了……嗯,从那以后我每个项目都先git init。
2.6 安装与配置Git
去git-scm.com下载安装包,一路默认安装就行。装完后打开终端:
git --version
能看到版本号就OK。
然后配置你的身份(提交代码时会用到):
git config --global user.name "你的名字"
git config --global user.email "你的邮箱"
这里用你GitHub的邮箱就行。我建议用真实姓名,别用网名,不然同事看了想打人。
2.7 初始化你的第一个仓库
咱们在桌面建个文件夹叫 aigc_course,然后:
cd ~/Desktop/aigc_course
git init
这就创建了一个本地仓库。以后你在这个文件夹里做的所有修改,Git都能帮你追踪。
常用的三个命令,先记住:
| 命令 | 作用 |
|---|---|
git add . |
把修改的文件加入暂存区 |
git commit -m "说明" |
提交一个版本 |
git log |
查看提交历史 |
注意:千万别把大文件(比如模型权重)提交到Git里。建议创建一个 .gitignore 文件,把 *.pth、*.pt、__pycache__/ 这些写进去。
2.8 VS Code:写代码的瑞士军刀
编辑器我推荐VS Code。免费、轻量、插件多。你想想看,一个编辑器能写Python、Markdown、JSON,还能连Git,多省事。
去code.visualstudio.com下载安装。装完后,咱们配几个必备插件:
- Python(微软官方)——语法高亮、代码补全、调试
- Jupyter——直接在VS Code里跑Notebook
- GitLens——在代码行旁边显示谁最后改的这行代码
- Markdown All in One——写文档时自动补全
安装方法:左侧点扩展图标(或按 Ctrl+Shift+X),搜索插件名,点安装。
2.9 把VS Code和Anaconda连起来
这一步很关键。打开VS Code,按 Ctrl+Shift+P,输入 Python: Select Interpreter,然后选择 aigc_env 里的Python。
怎么确认选对了?在VS Code底部状态栏,能看到Python版本和虚拟环境名。显示 Python 3.10.0 ('aigc_env': conda) 就对了。
避坑指南:我曾经遇到过选了环境但终端里还是系统Python的情况。解决办法:在VS Code里打开一个新终端(Ctrl+Shift+`),它会自动激活你选的那个环境。
2.10 测试一下:跑个Hello World
在 aigc_course 文件夹里新建一个 hello.py:
import torch
print(f"PyTorch版本: {torch.__version__}")
print("AIGC开发环境配置成功!")
右键 -> Run Python File in Terminal。看到输出就说明一切正常。
到这里,你的开发环境就搭好了。后面每一章,咱们都会在这个环境里写代码、跑模型。记住:每次打开VS Code,先确认右下角的环境是 aigc_env,不然跑不通别来找我。
下一章,咱们开始写第一个AIGC程序——用Hugging Face的pipeline做文本生成。到时候你就知道,环境配好了,写代码有多爽。