1. MapTR本地训练与测试环境搭建实录
MapTR作为当前热门的自动驾驶场景理解框架,其本地化部署过程往往成为工程师们的第一个挑战。上周我在Ubuntu 20.04系统上从零开始配置MapTR环境时,遇到了包括CUDA版本冲突、依赖项缺失在内的7类典型问题。本文将详细记录这些"坑"的填平过程,特别适合刚接触MapTR需要快速搭建本地实验环境的开发者参考。
需要模型API调用? 免费领10W Token,多模型网关一键接入 Claude、DeepSeek 等主流模型。
2. 环境准备阶段的典型问题排查
2.1 CUDA与PyTorch版本匹配陷阱
在安装PyTorch时最容易踩的坑就是CUDA版本不匹配。官方文档建议的torch==1.9.0+cu111组合在我的RTX 3090上直接报错:
bash复制RuntimeError: CUDA error: no kernel image is available for execution on the device
经过实测发现,30系显卡必须使用CUDA 11.3以上版本。最终采用的组合方案是:
- CUDA 11.6
- torch==1.12.1+cu116
- torchvision==0.13.1+cu116
重要提示:安装前务必通过
nvidia-smi和nvcc --version确认驱动版本与CUDA Toolkit版本的一致性。我遇到过驱动显示11.6但工具链仍是11.1的情况,导致编译时出现难以排查的段错误。
2.2 依赖项冲突的解决方案
MapTR的requirements.txt包含mmdetection等大型框架,直接pip安装极易引发依赖地狱。推荐使用conda创建独立环境:
bash复制conda create -n maptr python=3.8 -y
conda activate maptr
pip install torch==1.12.1+cu116 --extra-index-url https://download.pytorch.org/whl/cu116
然后手动安装mmcv-full时需要注意:
bash复制pip install mmcv-full==1.6.0 -f https://download.openmmlab.com/mmcv/dist/cu116/torch1.12.0/
