Orion的硬件需求根据使用场景有所不同:
- 基础推理:
- 推荐NVIDIA A100显卡(32GB显存)
- 最低要求RTX 3090(24GB显存)
- 训练需求:需要80GB以上显存的GPU(如A100 80GB或H100)
- 存储空间:
- 预训练模型约15GB
- Bench2Drive完整数据集需100GB+空间
注意事项:
- 闭环仿真测试需额外计算资源运行CARLA
- FP16精度模式可降低显存占用,但可能影响结果精度
- 未来版本可能通过模型压缩技术降低硬件门槛
本答案来源于文章《Orion:小米开源的端到端自动驾驶推理与规划框架》