Orion的安装需要以下步骤:
- 环境准备:
- Ubuntu 20.04系统
- Python 3.8(通过
python3 --version
验证) - Git和Conda(用于虚拟环境管理)
- 创建虚拟环境:
conda create -n orion python=3.8 -y
- 克隆代码库:
git clone https://github.com/xiaomi-mlab/Orion.git
- 安装PyTorch:需CUDA 11.8支持的PyTorch 2.4.1版本
- 下载预训练模型:从Hugging Face和OmniDrive获取2D语言模型与视觉编码器权重,存放于
ckpts
目录
完成安装后,可通过运行./adzoo/orion/orion_dist_eval.sh
脚本验证环境配置。
本答案来源于文章《Orion:小米开源的端到端自动驾驶推理与规划框架》