环境准备:Python版本要求、虚拟环境搭建、依赖包安装与验证
说实话,搭建开发环境这件事,看着简单,但坑是真不少。我见过太多人卡在这一步,一卡就是半天。咱们今天就把这事彻底捋清楚。
Python版本要求
Alpha生成框架对Python版本有明确要求。我个人习惯用3.10.x系列,为什么?因为3.11之后有些库的兼容性还没完全跟上,3.9又太老了。
具体来说:
- 最低要求:Python 3.9.0
- 推荐版本:Python 3.10.0 - 3.10.11
- 最高支持:Python 3.11.x(部分功能可能受限)
⚠️ 注意:别用Python 3.12!我在项目中遇到过,有些底层依赖包还没更新,装上去直接报错。你想想看,折腾半天结果发现是版本问题,多冤。
怎么检查你当前的Python版本?打开终端,敲下面这行:
python --version
# 或者
python3 --version
如果版本不对,去官网下载对应的安装包就行。嗯,这里要注意,Windows用户记得勾选「Add Python to PATH」,不然后面有你受的。
虚拟环境搭建
为什么要用虚拟环境?说白了,就是为了隔离。每个项目有自己的一套依赖,互不干扰。我曾经在一个机器上同时维护三个项目,没用虚拟环境,结果A项目升级了某个库,B项目直接崩了。从那以后,我每个项目必建虚拟环境。
搭建方式有两种,我推荐第二种:
方式一:venv(Python内置)
# 创建虚拟环境
python -m venv alpha_env
# 激活(Windows)
alpha_env\Scripts\activate
# 激活(Mac/Linux)
source alpha_env/bin/activate
方式二:conda(我推荐)
为什么推荐conda?因为它不光管Python包,还能管其他底层库。做AI相关项目,经常需要装一些C++编译的库,conda处理这些比pip强太多。
# 创建虚拟环境
conda create -n alpha_env python=3.10
# 激活
conda activate alpha_env
💡 小技巧:给环境起名字时,我习惯用项目名+环境类型,比如 alpha_env、alpha_dev。这样时间长了也不会搞混。
依赖包安装与验证
环境搭好了,接下来装依赖包。Alpha生成框架的核心依赖都在 requirements.txt 里。你想想看,要是手动一个个装,得装到什么时候?
直接一条命令搞定:
pip install -r requirements.txt
requirements.txt 的内容大致长这样:
torch>=2.0.0
transformers>=4.30.0
datasets>=2.12.0
accelerate>=0.20.0
peft>=0.4.0
bitsandbytes>=0.39.0
wandb>=0.15.0
tqdm>=4.65.0
numpy>=1.24.0
pandas>=2.0.0
装完之后,一定要验证。怎么验证?写个简单的测试脚本:
import torch
import transformers
import datasets
print(f"PyTorch版本: {torch.__version__}")
print(f"Transformers版本: {transformers.__version__}")
print(f"Datasets版本: {datasets.__version__}")
# 检查CUDA是否可用
if torch.cuda.is_available():
print(f"CUDA可用,GPU型号: {torch.cuda.get_device_name(0)}")
else:
print("CUDA不可用,将使用CPU训练(会很慢)")
🔑 关键验证点:
- 所有包都能正常导入,不报错
- 版本号与 requirements.txt 一致
- GPU能正常识别(如果有NVIDIA显卡)
如果遇到报错,别慌。我总结了几种常见情况:
| 错误类型 | 原因 | 解决办法 |
|---|---|---|
| ModuleNotFoundError | 包没装上 | 检查网络,重新 pip install |
| VersionConflict | 版本冲突 | 用 conda 重新创建环境,指定版本 |
| CUDA error | 驱动或CUDA版本不对 | 更新NVIDIA驱动,安装对应CUDA toolkit |
⚠️ 避坑指南:我曾经在Windows上装bitsandbytes,折腾了整整一下午。后来发现,Windows版需要从源码编译。建议Windows用户直接用WSL2,省心很多。
最后,咱们用一张图来总结整个环境准备流程:
环境准备这块,看着步骤多,其实熟练了五分钟就能搞定。关键是别急,一步一步来。我刚开始做的时候也踩过不少坑,但只要你按照上面的流程走,基本不会出大问题。
好了,环境搭好了,咱们就可以正式开始玩Alpha生成框架了。