第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 换成 cu118cu121(对应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做文本生成。到时候你就知道,环境配好了,写代码有多爽。