指南
Geneval MMCV 设置
Geneval/OpenMMLab workflow 所需的可选 MMCV 与 MMDetection 安装说明。
这组依赖是可选的。只有当 workflow 需要 Geneval/OpenMMLab stack 时才安装。
为什么单独安装
mmcv 和 mmdet 有 CUDA 编译约束,不是 UniRL 核心训练 pipeline 的必需依赖。
推荐版本
mmcvtag:v1.7.2mmdetectionbranch:2.x,验证环境中常见版本是2.28.2
安装步骤
示例使用 ~/mmgrpo 作为工作目录:
pip install -U openmim
mim install mmengine
cd ~/mmgrpo
git clone https://github.com/open-mmlab/mmcv.git
cd mmcv
git checkout tags/v1.7.2
MMCV_WITH_OPS=1 FORCE_CUDA=1 pip install -e . -v
cd ..
git clone https://github.com/open-mmlab/mmdetection.git
cd mmdetection
git checkout 2.x
pip install -e . -v
cd ..Sanity Check
python -c "import mmcv, mmdet; print('mmcv', mmcv.__version__); print('mmdet', mmdet.__version__); from mmcv.ops import nms; print('mmcv.ops ok')"期望结果:
mmcv为1.7.2mmdet为2.x版本mmcv.ops ok正常打印