环境准备: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,省心很多。

最后,咱们用一张图来总结整个环境准备流程:

环境准备核心流程 步骤一 检查Python版本 步骤二 创建虚拟环境 步骤三 安装依赖包 验证是否通过? ✅ 通过 → 开始开发 ❌ 不通过 → 排查问题 常见问题:版本冲突、CUDA不兼容、Windows编译失败

环境准备这块,看着步骤多,其实熟练了五分钟就能搞定。关键是别急,一步一步来。我刚开始做的时候也踩过不少坑,但只要你按照上面的流程走,基本不会出大问题。

好了,环境搭好了,咱们就可以正式开始玩Alpha生成框架了。


专注资料整理