1. 环境准备与版本控制
在开始安装MMDetection3D之前,必须严格把控环境配置。这个环节往往被新手忽视,但实际项目中90%的安装问题都源于环境版本冲突。我建议采用"虚拟环境+版本锁定"的方案,这是经过多个工业级项目验证的最佳实践。
1.1 Python环境配置
Python版本的选择不是随意的,必须遵循以下约束链:
- MMDetection3D依赖的numba==0.53.0
- numba 0.53.0最高支持Python 3.9
- 因此Python版本必须≤3.9
推荐使用conda创建隔离环境(假设环境名为mm3d-env):
bash复制conda create -n mm3d-env python=3.8 -y
conda activate mm3d-env
注意:虽然Python 3.9也符合要求,但实测发现部分依赖库在3.9下有兼容性问题。3.8是更稳妥的选择
1.2 CUDA与PyTorch匹配
GPU环境需要特别注意CUDA版本与PyTorch的对应关系。根据官方文档和实测结果,推荐以下组合:
| PyTorch版本 | 推荐CUDA版本 | 适用显卡架构 |
|---|---|---|
| 1.11.0 | 11.3/11.5 | Turing/Ampere |
| 1.10.2 | 11.3 | Volta+ |
安装命令示例(对应CUDA 11.5):
bash复制pip install torch==1.11.0 torchvision==0.12.0 --extra-index-url https://download.pytorch.org/whl/cu115
验证安装成功的标准检查点:
python复制import torch
print(torch.__version__) # 应输出1.11.0
print(torch.cuda.is_available()) # 必须返回True
print(torch.version.cuda) # 应显示11.x
需要模型API调用? 免费领10W Token,多模型网关一键接入 Claude、DeepSeek 等主流模型。
