1. 环境准备与VLLM安装全流程
在Linux系统上部署大语言模型需要严谨的环境配置,我以Ubuntu 22.04 LTS为例,详细记录整个部署过程。这个方案经过多次实际验证,能稳定运行Qwen、LLaMA等主流开源大模型。
1.1 基础环境检查
首先确认系统已安装NVIDIA驱动和CUDA工具包。这是GPU加速的基础条件:
bash复制nvidia-smi # 查看驱动版本和GPU状态
nvcc --version # 查看CUDA版本
注意:CUDA版本需要与后续安装的PyTorch版本严格匹配。我推荐使用CUDA 12.1,因其对新显卡的兼容性更好。
如果缺少驱动,可通过官方方式安装:
bash复制sudo apt install nvidia-driver-535
1.2 Python虚拟环境搭建
为避免依赖冲突,使用conda创建独立环境。Python版本选择3.10,这是目前大模型生态兼容性最好的版本:
bash复制conda create -n vllm_env python=3.10 -y
conda activate vllm_env
验证环境是否生效:
bash复制which python # 应显示conda环境路径
python --version # 确认是3.10.x
需要模型API调用? 免费领10W Token,多模型网关一键接入 Claude、DeepSeek 等主流模型。
2. 核心组件安装与验证
2.1 VLLM框架安装
VLLM是一个专为LLM推理优化的服务框架,支持连续批处理和PagedAttention等关键技术:
bash复制pip install vllm
安装后验证:
bash复制python -c "import vllm; print(vllm.__version__)"
常见问题:
- 如果报错
GLIBCXX缺失,需升级系统GCC:bash复制sudo apt install libstdc++6
2.2 PyTorch与CUDA适配
安装与CUDA 12.1匹配的PyTorch版本:
bash复制pip3 install torch torchvision torchaudio --index-url https://do
