什么是 OpenPCDet?
OpenPCDet 是一个开源的 3D 目标检测工具箱,支持多种点云模型(如 PointRCNN、Part-A2、PV-RCNN 等)。 虽然官方主要支持 Linux,但通过适当配置,也可在 Windows 上成功安装和运行。
前置要求
- Windows 10 或 Windows 11
- Python 3.7 - 3.9(推荐 3.8)
- PyTorch ≥ 1.8(需匹配 CUDA 版本)
- CUDA Toolkit(建议 11.1 - 11.8)
- Visual Studio 2019(含 C++ 构建工具)
- Git
安装步骤
1. 安装 Visual Studio 构建工具
下载并安装 Microsoft C++ Build Tools,确保勾选:
- C++ CMake 工具
- Windows 10 SDK
- C++ ATL(可选但推荐)
2. 创建 Python 虚拟环境(推荐)
python -m venv openpcdet-env
openpcdet-env\Scripts\activate
3. 安装 PyTorch
访问 PyTorch 官网,选择对应 CUDA 版本,例如:
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
4. 克隆 OpenPCDet 仓库
git clone https://github.com/open-mmlab/OpenPCDet.git
cd OpenPCDet
5. 安装依赖
pip install -r requirements.txt
6. 编译 CUDA 扩展(关键步骤)
在 Windows 上需修改部分源码以兼容 MSVC 编译器。建议使用以下命令:
python setup.py develop
注意: 若遇到编译错误,请确保:
- 使用 PowerShell 或 CMD(非 Git Bash)
- 已设置正确的 CUDA_PATH 环境变量
- Visual Studio 的开发人员命令提示符已正确配置
常见问题(FAQ)
Q1: 编译时报错 “error: Microsoft Visual C++ 14.0 is required”
请安装 Visual Studio 2019 或 Build Tools,并重启终端。
Q2: 出现 “nvcc not recognized” 错误
检查 CUDA 是否正确安装,并将 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.x\bin 加入系统 PATH。
Q3: 能否使用 WSL2 安装?
可以!WSL2(Ubuntu)是更推荐的方式,避免 Windows 编译兼容性问题。
验证安装
运行以下命令测试是否安装成功:
python -c "import pcdet; print('OpenPCDet 导入成功!')"
若无报错,则说明安装成功。