依赖
下表为 MMYOLO 和 MMEngine, MMCV, MMDetection 依赖库的版本要求，请安装正确的版本以避免安装问题。
| MMYOLO version | MMDetection version | MMEngine version | MMCV version |
|---|---|---|---|
| main | mmdet>=3.0.0, <3.1.0 | mmengine>=0.7.1, <1.0.0 | mmcv>=2.0.0rc4, <2.1.0 |
| 0.6.0 | mmdet>=3.0.0, <3.1.0 | mmengine>=0.7.1, <1.0.0 | mmcv>=2.0.0rc4, <2.1.0 |
| 0.5.0 | mmdet>=3.0.0rc6, <3.1.0 | mmengine>=0.6.0, <1.0.0 | mmcv>=2.0.0rc4, <2.1.0 |
| 0.4.0 | mmdet>=3.0.0rc5, <3.1.0 | mmengine>=0.3.1, <1.0.0 | mmcv>=2.0.0rc0, <2.1.0 |
| 0.3.0 | mmdet>=3.0.0rc5, <3.1.0 | mmengine>=0.3.1, <1.0.0 | mmcv>=2.0.0rc0, <2.1.0 |
| 0.2.0 | mmdet>=3.0.0rc3, <3.1.0 | mmengine>=0.3.1, <1.0.0 | mmcv>=2.0.0rc0, <2.1.0 |
| 0.1.3 | mmdet>=3.0.0rc3, <3.1.0 | mmengine>=0.3.1, <1.0.0 | mmcv>=2.0.0rc0, <2.1.0 |
| 0.1.2 | mmdet>=3.0.0rc2, <3.1.0 | mmengine>=0.3.0, <1.0.0 | mmcv>=2.0.0rc0, <2.1.0 |
| 0.1.1 | mmdet==3.0.0rc1 | mmengine>=0.1.0, <0.2.0 | mmcv>=2.0.0rc0, <2.1.0 |
| 0.1.0 | mmdet==3.0.0rc0 | mmengine>=0.1.0, <0.2.0 | mmcv>=2.0.0rc0, <2.1.0 |
本节中，我们将演示如何用 PyTorch 准备一个环境。
MMYOLO 支持在 Linux，Windows 和 macOS 上运行。它的基本环境依赖为：
- Python 3.7+
- PyTorch 1.7+
- CUDA 9.2+
- GCC 5.4+
如果你对 PyTorch 有经验并且已经安装了它，你可以直接跳转到下一小节。否则，你可以按照下述步骤进行准备
步骤 0. 从 官方网站 下载并安装 Miniconda。
步骤 1. 创建并激活一个 conda 环境。
conda create -n mmyolo python=3.8 -y
conda activate mmyolo
步骤 2. 基于 PyTorch 官方说明 安装 PyTorch。
在 GPU 平台上：
conda install pytorch torchvision -c pytorch
在 CPU 平台上:
conda install pytorch torchvision cpuonly -c pytorch
步骤 3. 验证 PyTorch 安装
python -c "import torch; print(torch.__version__); print(torch.cuda.is_available())"
如果是在 GPU 平台上，那么会打印版本信息和 True 字符，否则打印版本信息和 False 字符。