UniRL 文档
指南

Geneval MMCV 设置

Geneval/OpenMMLab workflow 所需的可选 MMCV 与 MMDetection 安装说明。

这组依赖是可选的。只有当 workflow 需要 Geneval/OpenMMLab stack 时才安装。

为什么单独安装

mmcvmmdet 有 CUDA 编译约束,不是 UniRL 核心训练 pipeline 的必需依赖。

推荐版本

  • mmcv tag: v1.7.2
  • mmdetection branch: 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')"

期望结果:

  • mmcv1.7.2
  • mmdet2.x 版本
  • mmcv.ops ok 正常打印

目录