📌 先定核心原则:环境搭建的 3 个铁律
环境隔离
:每个项目用独立 Conda 环境,绝不污染全局 Python,避免版本冲突;
版本锁死
:PyTorch、CUDA、Transformers 版本严格匹配,绝不盲目装最新版;
先验后装
:先确认显卡、驱动、CUDA 版本,再装对应依赖,不瞎装导致不兼容。
🔧 分步实操:从 0 到 1 搭建全流程
1. 前置准备:先确认你的硬件与驱动
确认显卡:NVIDIA RTX 系列(必须支持 CUDA,AMD 显卡不适用这套方案);
确认驱动版本:打开 NVIDIA 控制面板 → 帮助 → 系统信息 → 组件,确认 CUDA 版本 ≥ 12.1;
提前安装 Anaconda/Miniconda:用 Miniconda 更轻量,官网下载安装,勾选「Add to PATH」。
2. 第一步:创建独立 Conda 环境(隔离核心)
# 创建环境,指定Python 3.10(最稳定,无版本冲突)conda create -n ai-dev python=3.10 -y# 激活环境,后续所有操作都必须在这个环境里执行conda activate ai-dev
环境名建议固定用ai-dev,避免和其他项目混淆;
Python 版本锁死 3.10,不要用 3.11/3.12,很多 AI 库兼容性差。
3. 第二步:安装适配显卡的 PyTorch(核心中的核心)
# 安装GPU版PyTorch,带CUDA支持pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121
import torchprint(torch.cuda.is_available()) # 必须输出True,否则就是CPU版,白装了print(torch.cuda.get_device_name(0)) # 输出你的显卡型号,比如GeForce RTX 4050 Laptop GPU
不要用conda install torch,大概率装到 CPU 版,必须用上面的 pip 命令;
不要盲目装最新版 PyTorch,2.3.0 版本最稳定,适配所有开源模型。
4. 第三步:安装全套 AI 依赖库
# 核心依赖,全是推理、微调必须的库pip install transformers datasets accelerate peft bitsandbytes sentencepiece matplotlib fastapi uvicorn
pip install transformers==4.40.0 datasets==2.18.0 accelerate==0.30.1 peft==0.10.0 bitsandbytes==0.43.1bitsandbytes
是量化核心库,Windows 版必须装 0.43.1 及以上,旧版本会报错;
安装时如果速度慢,加国内镜像源:-i https://pypi.tuna.tsinghua.edu.cn/simple。
5. 第四步:验证环境可用(跑个小测试)
# 测试代码,直接运行,无报错、正常输出即为成功from transformers import AutoTokenizer, AutoModelForCausalLM, BitsAndBytesConfig# 4bit量化配置,适配6G显存bnb_config = BitsAndBytesConfig(load_in_4bit=True,bnb_4bit_quant_type="nf4",bnb_4bit_compute_dtype="float16")print("环境验证通过,所有依赖正常!")
⚠️ 高频踩坑点(实测避坑,90% 的人都卡过)
PyTorch 装完还是用 CPU:原因:装了 CPU 版 PyTorch,或者 CUDA 驱动版本太低。解决:卸载重装,必须用上面带 CUDA 索引的 pip 命令,升级 NVIDIA 驱动到最新版。
bitsandbytes 在 Windows 报错:原因:旧版本不支持 Windows,或者没有开启管理员权限。解决:升级到 0.43.1 及以上版本,终端用管理员模式运行。
依赖版本冲突:原因:全局 Python 污染,或者装了多个版本的库。解决:删除环境重建,严格按上面的步骤,全程在ai-dev环境里操作,不装多余依赖。
模型下载慢、卡断:解决:配置 HuggingFace 镜像,终端执行:
set HF_ENDPOINT=https://hf-mirror.com🔧 环境固化:一次搭建,永久复用
# 导出环境配置conda env export --name ai-dev > ai-dev-env.yml# 后续恢复环境,一行命令搞定conda env create -f ai-dev-env.yml
夜雨聆风