1. 项目概述
OpenEMMA是一个基于多模态大语言模型的自动驾驶场景理解与轨迹预测框架,结合了视觉感知与运动预测能力。本文将详细介绍如何在本地环境部署OpenEMMA项目,并集成Qwen2.5-VL-3B-AWQ模型进行高效推理。
需要模型API调用? 免费领10W Token,多模型网关一键接入 Claude、DeepSeek 等主流模型。
2. 环境准备
2.1 硬件与系统要求
对于深度学习项目,合理的硬件配置是确保模型高效运行的基础。以下是经过实测验证的推荐配置:
| 组件 | 推荐配置 | 说明 |
|---|---|---|
| GPU | NVIDIA 8GB+ 显存 | RTX 4060/3060/4090均可流畅运行 |
| CUDA | 12.x | 需与PyTorch版本兼容 |
| 操作系统 | Ubuntu 20.04/22.04 | 其他Linux发行版也可运行 |
| Python版本 | 3.10 | 测试中最稳定的Python版本 |
提示:虽然项目理论上支持CUDA 11.8,但实测CUDA 12.x环境下性能更优,建议优先选择。
2.2 Conda环境配置
使用Conda创建隔离的Python环境是管理项目依赖的最佳实践:
bash复制# 创建Python 3.10环境
conda create -n openemma python=3.10 -y
conda activate openemma
# 升级基础工具链
pip install --upgrade pip setuptools wheel
2.3 PyTorch安装
PyTorch的版本需要与CUDA版本严格匹配。首先确认CUDA版本:
bash复制nvidia-smi
根据CUDA版本选择对应的PyTorch安装命令:
bash复制# CUDA 12.x版本(推荐)
pip install torch==2.10.0
