diff --git a/README.md b/README.md index 1a6541a689a..407c8672e81 100644 --- a/README.md +++ b/README.md @@ -20,6 +20,7 @@ [![docs](https://img.shields.io/badge/docs-latest-blue)](https://mmcv.readthedocs.io/en/latest/) +[![platform](https://img.shields.io/badge/platform-Linux%7CWindows%7CmacOS-blue)](https://mmcv.readthedocs.io/en/latest/get_started/installation.html) [![PyPI - Python Version](https://img.shields.io/pypi/pyversions/mmcv)](https://pypi.org/project/mmcv/) [![PyPI](https://img.shields.io/pypi/v/mmcv)](https://pypi.org/project/mmcv) [![badge](https://github.com/open-mmlab/mmcv/workflows/build/badge.svg)](https://github.com/open-mmlab/mmcv/actions) @@ -30,39 +31,17 @@ English | [简体中文](README_zh-CN.md) ## Introduction -MMCV is a foundational library for computer vision research and supports many -research projects as below: +MMCV is a foundational library for computer vision research and it provides the following functionalities: -- [MIM](https://github.com/open-mmlab/mim): MIM installs OpenMMLab packages. -- [MMClassification](https://github.com/open-mmlab/mmclassification): OpenMMLab image classification toolbox and benchmark. -- [MMDetection](https://github.com/open-mmlab/mmdetection): OpenMMLab detection toolbox and benchmark. -- [MMDetection3D](https://github.com/open-mmlab/mmdetection3d): OpenMMLab's next-generation platform for general 3D object detection. -- [MMRotate](https://github.com/open-mmlab/mmrotate): OpenMMLab rotated object detection toolbox and benchmark. -- [MMSegmentation](https://github.com/open-mmlab/mmsegmentation): OpenMMLab semantic segmentation toolbox and benchmark. -- [MMOCR](https://github.com/open-mmlab/mmocr): OpenMMLab text detection, recognition, and understanding toolbox. -- [MMPose](https://github.com/open-mmlab/mmpose): OpenMMLab pose estimation toolbox and benchmark. -- [MMHuman3D](https://github.com/open-mmlab/mmhuman3d): OpenMMLab 3D human parametric model toolbox and benchmark. -- [MMSelfSup](https://github.com/open-mmlab/mmselfsup): OpenMMLab self-supervised learning toolbox and benchmark. -- [MMRazor](https://github.com/open-mmlab/mmrazor): OpenMMLab model compression toolbox and benchmark. -- [MMFewShot](https://github.com/open-mmlab/mmfewshot): OpenMMLab fewshot learning toolbox and benchmark. -- [MMAction2](https://github.com/open-mmlab/mmaction2): OpenMMLab's next-generation action understanding toolbox and benchmark. -- [MMTracking](https://github.com/open-mmlab/mmtracking): OpenMMLab video perception toolbox and benchmark. -- [MMFlow](https://github.com/open-mmlab/mmflow): OpenMMLab optical flow toolbox and benchmark. -- [MMEditing](https://github.com/open-mmlab/mmediting): OpenMMLab image and video editing toolbox. -- [MMGeneration](https://github.com/open-mmlab/mmgeneration): OpenMMLab image and video generative models toolbox. -- [MMDeploy](https://github.com/open-mmlab/mmdeploy): OpenMMLab model deployment framework. +- [Universal IO APIs](https://mmcv.readthedocs.io/en/latest/understand_mmcv/io.html) +- [Image/Video processing](https://mmcv.readthedocs.io/en/latest/understand_mmcv/data_process.html) +- [Image and annotation visualization](https://mmcv.readthedocs.io/en/latest/understand_mmcv/visualization.html) +- [Useful utilities (progress bar, timer, ...)](https://mmcv.readthedocs.io/en/latest/understand_mmcv/utils.html) +- [PyTorch runner with hooking mechanism](https://mmcv.readthedocs.io/en/latest/understand_mmcv/runner.html) +- [Various CNN architectures](https://mmcv.readthedocs.io/en/latest/understand_mmcv/cnn.html) +- [High-quality implementation of common CPU and CUDA ops](https://mmcv.readthedocs.io/en/latest/understand_mmcv/ops.html) -It provides the following functionalities. - -- Universal IO APIs -- Image/Video processing -- Image and annotation visualization -- Useful utilities (progress bar, timer, ...) -- PyTorch runner with hooking mechanism -- Various CNN architectures -- High-quality implementation of common CUDA ops - -It supports the following systems. +It supports the following systems: - Linux - Windows @@ -76,181 +55,83 @@ Note: MMCV requires Python 3.6+. There are two versions of MMCV: -- **mmcv-full**: comprehensive, with full features and various CUDA ops out of box. It takes longer time to build. -- **mmcv**: lite, without CUDA ops but all other features, similar to mmcv\<1.0.0. It is useful when you do not need those CUDA ops. +- **mmcv-full**: comprehensive, with full features and various CPU and CUDA ops out of the box. It takes longer time to build. +- **mmcv**: lite, without CPU and CUDA ops but all other features, similar to mmcv\<1.0.0. It is useful when you do not need those CUDA ops. **Note**: Do not install both versions in the same environment, otherwise you may encounter errors like `ModuleNotFound`. You need to uninstall one before installing the other. `Installing the full version is highly recommended if CUDA is available`. -a. Install the full version. - -Before installing mmcv-full, make sure that PyTorch has been successfully installed following the [official guide](https://pytorch.org/). +### Install mmcv-full -We provide pre-built mmcv packages (recommended) with different PyTorch and CUDA versions to simplify the building for **Linux and Windows systems**. In addition, you can run [check_installation.py](.dev_scripts/check_installation.py) to check the installation of mmcv-full after running the installation commands. +Before installing mmcv-full, make sure that PyTorch has been successfully installed following the [PyTorch official installation guide](https://github.com/pytorch/pytorch#installation). -i. Install the latest version. +The command to install mmcv-full on Linux or Windows platforms is as follows (if your system is macOS, please refer to [build mmcv-full from source](https://mmcv.readthedocs.io/en/latest/get_started/build.html#macos-mmcv-full)) -The rule for installing the latest `mmcv-full` is as follows: - -```shell -pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/{cu_version}/{torch_version}/index.html +```bash +pip install -U openmim +mim install mmcv-full ``` -Please replace `{cu_version}` and `{torch_version}` in the url to your desired one. For example, -to install the latest `mmcv-full` with `CUDA 11.1` and `PyTorch 1.9.0`, use the following command: +If you need to specify the version of mmcv-full, you can use the following command -```shell -pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/cu111/torch1.9.0/index.html +```bash +mim install mmcv-full==1.5.0 ``` -**Note**: mmcv-full is only compiled on PyTorch 1.x.0 because the compatibility usually holds between 1.x.0 and 1.x.1. If your PyTorch version is 1.x.1, you can install mmcv-full compiled with PyTorch 1.x.0 and it usually works well. For example, if your PyTorch version is 1.8.1 and CUDA version is 11.1, you can use the following command to install mmcv-full. - -```shell -pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/cu111/torch1.8.0/index.html -``` +If you find that the above installation command does not use a pre-built package ending with `.whl` but a source package ending with `.tar.gz`, you may not have a pre-build package corresponding to the PyTorch or CUDA or mmcv-full version, in which case you can [build mmcv-full from source](https://mmcv.readthedocs.io/en/latest/get_started/build.html). -For more details, please refer the the following tables and delete `=={mmcv_version}`. +
+Installation log using pre-built packages -ii. Install a specified version. +Looking in links: https://download.openmmlab.com/mmcv/dist/cu102/torch1.8.0/index.html
+Collecting mmcv-full
+Downloading https://download.openmmlab.com/mmcv/dist/cu102/torch1.8.0/mmcv_full-1.6.1-cp38-cp38-manylinux1_x86_64.whl -The rule for installing a specified `mmcv-full` is as follows: +
-```shell -pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/{cu_version}/{torch_version}/index.html -``` +
+Installation log using source packages -First of all, please refer to the Releases and replace `{mmcv_version}` a specified one. e.g. `1.3.9`. -Then replace `{cu_version}` and `{torch_version}` in the url to your desired versions. For example, -to install `mmcv-full==1.3.9` with `CUDA 11.1` and `PyTorch 1.9.0`, use the following command: +Looking in links: https://download.openmmlab.com/mmcv/dist/cu102/torch1.8.0/index.html
+Collecting mmcv-full==1.6.0
+Downloading mmcv-full-1.6.0.tar.gz -```shell -pip install mmcv-full==1.3.9 -f https://download.openmmlab.com/mmcv/dist/cu111/torch1.9.0/index.html -``` +
-For more details, please refer the the following tables. - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
CUDA torch 1.11torch 1.10torch 1.9torch 1.8torch 1.7torch 1.6torch 1.5
11.5
install
pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cu115/torch1.11.0/index.html
11.3
install
pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cu113/torch1.11.0/index.html
install
pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cu113/torch1.10.0/index.html
11.1
install
pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cu111/torch1.10.0/index.html
install
pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cu111/torch1.9.0/index.html
install
pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cu111/torch1.8.0/index.html
11.0
install
pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cu110/torch1.7.0/index.html
10.2
install
pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cu102/torch1.11.0/index.html
install
pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cu102/torch1.10.0/index.html
install
pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cu102/torch1.9.0/index.html
install
pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cu102/torch1.8.0/index.html
install
pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cu102/torch1.7.0/index.html
install
pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cu102/torch1.6.0/index.html
install
pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cu102/torch1.5.0/index.html
10.1
install
 pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cu101/torch1.8.0/index.html
install
 pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cu101/torch1.7.0/index.html
install
 pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cu101/torch1.6.0/index.html
install
 pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cu101/torch1.5.0/index.html
9.2
install
 pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cu92/torch1.7.0/index.html
install
 pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cu92/torch1.6.0/index.html
install
 pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cu92/torch1.5.0/index.html
cpu
install
 pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cpu/torch1.11.0/index.html
install
 pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cpu/torch1.10.0/index.html
install
 pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cpu/torch1.9.0/index.html
install
 pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cpu/torch1.8.0/index.html
install
 pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cpu/torch1.7.0/index.html
install
 pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cpu/torch1.6.0/index.html
install
 pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cpu/torch1.5.0/index.html
- -**Note**: The pre-built packages provided above do not include all versions of mmcv-full, you can click on the corresponding links to see the supported versions. For example, you can click [cu102-torch1.8.0](https://download.openmmlab.com/mmcv/dist/cu102/torch1.8.0/index.html) and you can see that `cu102-torch1.8.0` only provides 1.3.0 and above versions of mmcv-full. In addition, We no longer provide `mmcv-full` pre-built packages compiled with `PyTorch 1.3 & 1.4` since v1.3.17. You can find previous versions that compiled with PyTorch 1.3 & 1.4 [here](./docs/en/get_started/previous_versions.md). The compatibility is still ensured in our CI, but we will discard the support of PyTorch 1.3 & 1.4 next year. - -**Note**: mmcv-full does not provide pre-built packages for `cu102-torch1.11` and `cu92-torch*` on Windows. - -Another way is to compile locally by running - -```python -pip install mmcv-full -``` +For more installation methods, please refer to the [Installation documentation](https://mmcv.readthedocs.io/en/latest/get_started/installation.html). -Note that the local compiling may take up to 10 mins. +### Install mmcv -b. Install the lite version. +If you need to use PyTorch-related modules, make sure PyTorch has been successfully installed in your environment by referring to the [PyTorch official installation guide](https://github.com/pytorch/pytorch#installation). -```python -pip install mmcv +```bash +pip install -U openmim +mim install mmcv ``` -c. Install full version with custom operators for onnxruntime - -- Check [here](docs/en/deployment/onnxruntime_op.md) for detailed instruction. +## Supported projects -If you would like to build MMCV from source, please refer to the [guide](https://mmcv.readthedocs.io/en/latest/get_started/build.html). +- [MIM](https://github.com/open-mmlab/mim): MIM installs OpenMMLab packages. +- [MMClassification](https://github.com/open-mmlab/mmclassification): OpenMMLab image classification toolbox and benchmark. +- [MMDetection](https://github.com/open-mmlab/mmdetection): OpenMMLab detection toolbox and benchmark. +- [MMDetection3D](https://github.com/open-mmlab/mmdetection3d): OpenMMLab's next-generation platform for general 3D object detection. +- [MMRotate](https://github.com/open-mmlab/mmrotate): OpenMMLab rotated object detection toolbox and benchmark. +- [MMSegmentation](https://github.com/open-mmlab/mmsegmentation): OpenMMLab semantic segmentation toolbox and benchmark. +- [MMOCR](https://github.com/open-mmlab/mmocr): OpenMMLab text detection, recognition, and understanding toolbox. +- [MMPose](https://github.com/open-mmlab/mmpose): OpenMMLab pose estimation toolbox and benchmark. +- [MMHuman3D](https://github.com/open-mmlab/mmhuman3d): OpenMMLab 3D human parametric model toolbox and benchmark. +- [MMSelfSup](https://github.com/open-mmlab/mmselfsup): OpenMMLab self-supervised learning toolbox and benchmark. +- [MMRazor](https://github.com/open-mmlab/mmrazor): OpenMMLab model compression toolbox and benchmark. +- [MMFewShot](https://github.com/open-mmlab/mmfewshot): OpenMMLab fewshot learning toolbox and benchmark. +- [MMAction2](https://github.com/open-mmlab/mmaction2): OpenMMLab's next-generation action understanding toolbox and benchmark. +- [MMTracking](https://github.com/open-mmlab/mmtracking): OpenMMLab video perception toolbox and benchmark. +- [MMFlow](https://github.com/open-mmlab/mmflow): OpenMMLab optical flow toolbox and benchmark. +- [MMEditing](https://github.com/open-mmlab/mmediting): OpenMMLab image and video editing toolbox. +- [MMGeneration](https://github.com/open-mmlab/mmgeneration): OpenMMLab image and video generative models toolbox. +- [MMDeploy](https://github.com/open-mmlab/mmdeploy): OpenMMLab model deployment framework. ## FAQ -If you face some installation issues, CUDA related issues or RuntimeErrors, -you may first refer to this [Frequently Asked Questions](https://mmcv.readthedocs.io/en/latest/faq.html). +If you face installation problems or runtime issues, you may first refer to this [Frequently Asked Questions](https://mmcv.readthedocs.io/zh_CN/latest/faq.html) to see if there is a solution. If the problem is still not solved, feel free to open an [issue](https://github.com/open-mmlab/mmcv/issues). ## Citation diff --git a/README_zh-CN.md b/README_zh-CN.md index 8c768c837ec..187236dbd87 100644 --- a/README_zh-CN.md +++ b/README_zh-CN.md @@ -20,6 +20,7 @@ [![docs](https://img.shields.io/badge/docs-latest-blue)](https://mmcv.readthedocs.io/zh_CN/latest/) +[![platform](https://img.shields.io/badge/platform-Linux%7CWindows%7CmacOS-blue)](https://mmcv.readthedocs.io/zh_CN/latest/get_started/installation.html) [![PyPI - Python Version](https://img.shields.io/pypi/pyversions/mmcv)](https://pypi.org/project/mmcv/) [![PyPI](https://img.shields.io/pypi/v/mmcv)](https://pypi.org/project/mmcv) [![badge](https://github.com/open-mmlab/mmcv/workflows/build/badge.svg)](https://github.com/open-mmlab/mmcv/actions) @@ -30,44 +31,23 @@ ## 简介 -MMCV 是一个面向计算机视觉的基础库,它支持了很多开源项目,例如: +MMCV 是一个面向计算机视觉的基础库,它提供了以下功能: -- [MIM](https://github.com/open-mmlab/mim): MIM 是 OpenMMlab 项目、算法、模型的统一入口 -- [MMClassification](https://github.com/open-mmlab/mmclassification): OpenMMLab 图像分类工具箱 -- [MMDetection](https://github.com/open-mmlab/mmdetection): OpenMMLab 目标检测工具箱 -- [MMDetection3D](https://github.com/open-mmlab/mmdetection3d): OpenMMLab 新一代通用 3D 目标检测平台 -- [MMRotate](https://github.com/open-mmlab/mmrotate): OpenMMLab 旋转框检测工具箱与测试基准 -- [MMSegmentation](https://github.com/open-mmlab/mmsegmentation): OpenMMLab 语义分割工具箱 -- [MMOCR](https://github.com/open-mmlab/mmocr): OpenMMLab 全流程文字检测识别理解工具箱 -- [MMPose](https://github.com/open-mmlab/mmpose): OpenMMLab 姿态估计工具箱 -- [MMHuman3D](https://github.com/open-mmlab/mmhuman3d): OpenMMLab 人体参数化模型工具箱与测试基准 -- [MMSelfSup](https://github.com/open-mmlab/mmselfsup): OpenMMLab 自监督学习工具箱与测试基准 -- [MMRazor](https://github.com/open-mmlab/mmrazor): OpenMMLab 模型压缩工具箱与测试基准 -- [MMFewShot](https://github.com/open-mmlab/mmfewshot): OpenMMLab 少样本学习工具箱与测试基准 -- [MMAction2](https://github.com/open-mmlab/mmaction2): OpenMMLab 新一代视频理解工具箱 -- [MMTracking](https://github.com/open-mmlab/mmtracking): OpenMMLab 一体化视频目标感知平台 -- [MMFlow](https://github.com/open-mmlab/mmflow): OpenMMLab 光流估计工具箱与测试基准 -- [MMEditing](https://github.com/open-mmlab/mmediting): OpenMMLab 图像视频编辑工具箱 -- [MMGeneration](https://github.com/open-mmlab/mmgeneration): OpenMMLab 图片视频生成模型工具箱 -- [MMDeploy](https://github.com/open-mmlab/mmdeploy): OpenMMLab 模型部署框架 +- [通用的 IO 接口](https://mmcv.readthedocs.io/zh_CN/latest/understand_mmcv/io.html) +- [图像和视频处理](https://mmcv.readthedocs.io/zh_CN/latest/understand_mmcv/data_process.html) +- [图像和标注结果可视化](https://mmcv.readthedocs.io/zh_CN/latest/understand_mmcv/visualization.html) +- [常用小工具(进度条,计时器等)](https://mmcv.readthedocs.io/zh_CN/latest/understand_mmcv/utils.html) +- [基于 PyTorch 的通用训练框架](https://mmcv.readthedocs.io/zh_CN/latest/understand_mmcv/runner.html) +- [多种 CNN 网络结构](https://mmcv.readthedocs.io/zh_CN/latest/understand_mmcv/cnn.html) +- [高质量实现的 CPU 和 CUDA 算子](https://mmcv.readthedocs.io/zh_CN/latest/understand_mmcv/ops.html) -MMCV 提供了如下众多功能: - -- 通用的 IO 接口 -- 图像和视频处理 -- 图像和标注结果可视化 -- 常用小工具(进度条,计时器等) -- 基于 PyTorch 的通用训练框架 -- 多种 CNN 网络结构 -- 高质量实现的常见 CUDA 算子 - -MMCV 支持以下的系统: +MMCV 支持多种平台,包括: - Linux - Windows - macOS -如想了解更多特性和使用,请参考[文档](http://mmcv.readthedocs.io/zh_CN/latest)。 +如想了解更多特性和用法,请参考[文档](http://mmcv.readthedocs.io/zh_CN/latest)。 提示: MMCV 需要 Python 3.6 以上版本。 @@ -75,178 +55,83 @@ MMCV 支持以下的系统: MMCV 有两个版本: -- **mmcv-full**: 完整版,包含所有的特性以及丰富的开箱即用的 CUDA 算子。注意完整版本可能需要更长时间来编译。 -- **mmcv**: 精简版,不包含 CUDA 算子但包含其余所有特性和功能,类似 MMCV 1.0 之前的版本。如果你不需要使用 CUDA 算子的话,精简版可以作为一个考虑选项。 - -**注意**: 请不要在同一个环境中安装两个版本,否则可能会遇到类似 `ModuleNotFound` 的错误。在安装一个版本之前,需要先卸载另一个。`如果CUDA可用,强烈推荐安装mmcv-full`。 +- **mmcv-full**: 完整版,包含所有的特性以及丰富的开箱即用的 CPU 和 CUDA 算子。 +- **mmcv**: 精简版,不包含 CPU 和 CUDA 算子但包含其余所有特性和功能,类似 MMCV 1.0 之前的版本。如果你不需要使用算子的话,精简版可以作为一个考虑选项。 -a. 安装完整版 +**注意**: 请不要在同一个环境中安装两个版本,否则可能会遇到类似 `ModuleNotFound` 的错误。在安装一个版本之前,需要先卸载另一个。`如果 CUDA 可用,强烈推荐安装 mmcv-full`。 -在安装 mmcv-full 之前,请确保 PyTorch 已经成功安装在环境中,可以参考 PyTorch [官方文档](https://pytorch.org/)。 +### 安装 mmcv-full -我们提供了 **Linux 和 Windows 平台** PyTorch 和 CUDA 版本组合的 mmcv-full 预编译包,可以大大简化用户安装编译过程。强烈推荐通过预编译包来安装。另外,安装完成后可以运行 [check_installation.py](.dev_scripts/check_installation.py) 脚本检查 mmcv-full 是否安装成功。 +在安装 mmcv-full 之前,请确保 PyTorch 已经成功安装在环境中,可以参考 [PyTorch 官方安装文档](https://github.com/pytorch/pytorch#installation)。 -i. 安装最新版本 +在 Linux 和 Windows 平台安装 mmcv-full 的命令如下(如果你的系统是 macOS,请参考[源码安装 mmcv-full](https://mmcv.readthedocs.io/zh_CN/latest/get_started/build.html#macos-mmcv-full)) -如下是安装最新版 `mmcv-full` 的命令 - -```shell -pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/{cu_version}/{torch_version}/index.html +```bash +pip install -U openmim +mim install mmcv-full ``` -请将链接中的 `{cu_version}` 和 `{torch_version}` 根据自身需求替换成实际的版本号,例如想安装和 `CUDA 11.1`、`PyTorch 1.9.0` 兼容的最新版 `mmcv-full`,使用如下替换过的命令 +如果需要指定 mmcv-full 的版本,可以使用以下命令 -```shell -pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/cu111/torch1.9.0/index.html +```bash +mim install mmcv-full==1.5.0 ``` -**注意**: PyTorch 在 1.x.0 和 1.x.1 之间通常是兼容的,故 mmcv-full 只提供 1.x.0 的编译包。如果你的 PyTorch 版本是 1.x.1,你可以放心地安装在 1.x.0 版本编译的 mmcv-full。例如,如果你的 PyTorch 版本是 1.8.1、CUDA 版本是 11.1,你可以使用以下命令安装 mmcv-full。 - -```shell -pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/cu111/torch1.8.0/index.html -``` +如果发现上述的安装命令没有使用预编译包(以 `.whl` 结尾)而是使用源码包(以 `.tar.gz` 结尾)安装,则有可能是我们没有提供和当前环境的 PyTorch 版本、CUDA 版本相匹配的 mmcv-full 预编译包,此时,你可以[源码安装 mmcv-full](https://mmcv.readthedocs.io/zh_CN/latest/get_started/build.html)。 -如果想知道更多 CUDA 和 PyTorch 版本的命令,可以参考下面的表格,将链接中的 `=={mmcv_version}` 删去即可。 +
+使用预编译包的安装日志 -ii. 安装特定的版本 +Looking in links: https://download.openmmlab.com/mmcv/dist/cu102/torch1.8.0/index.html
+Collecting mmcv-full
+Downloading https://download.openmmlab.com/mmcv/dist/cu102/torch1.8.0/mmcv_full-1.6.1-cp38-cp38-manylinux1_x86_64.whl -如下是安装特定版本 `mmcv-full` 的命令 +
-```shell -pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/{cu_version}/{torch_version}/index.html -``` +
+使用源码包的安装日志 -首先请参考版本发布信息找到想要安装的版本号,将 `{mmcv_version}` 替换成该版本号,例如 `1.3.9`。 -然后将链接中的 `{cu_version}` 和 `{torch_version}` 根据自身需求替换成实际的版本号,例如想安装和 `CUDA 11.1`、`PyTorch 1.9.0` 兼容的 `mmcv-full` 1.3.9 版本,使用如下替换过的命令 +Looking in links: https://download.openmmlab.com/mmcv/dist/cu102/torch1.8.0/index.html
+Collecting mmcv-full==1.6.0
+Downloading mmcv-full-1.6.0.tar.gz -```shell -pip install mmcv-full==1.3.9 -f https://download.openmmlab.com/mmcv/dist/cu111/torch1.9.0/index.html -``` +
-对于更多的 PyTorch 和 CUDA 版本组合,请参考下表: - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
CUDA torch 1.11torch 1.10torch 1.9torch 1.8torch 1.7torch 1.6torch 1.5
11.5
安装
pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cu115/torch1.11.0/index.html
11.3
安装
pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cu113/torch1.11.0/index.html
安装
pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cu113/torch1.10.0/index.html
11.1
安装
pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cu111/torch1.10.0/index.html
安装
pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cu111/torch1.9.0/index.html
安装
pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cu111/torch1.8.0/index.html
11.0
安装
pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cu110/torch1.7.0/index.html
10.2
安装
pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cu102/torch1.11.0/index.html
安装
pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cu102/torch1.10.0/index.html
安装
pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cu102/torch1.9.0/index.html
安装
pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cu102/torch1.8.0/index.html
安装
pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cu102/torch1.7.0/index.html
安装
pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cu102/torch1.6.0/index.html
安装
pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cu102/torch1.5.0/index.html
10.1
安装
 pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cu101/torch1.8.0/index.html
安装
 pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cu101/torch1.7.0/index.html
安装
 pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cu101/torch1.6.0/index.html
安装
 pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cu101/torch1.5.0/index.html
9.2
安装
 pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cu92/torch1.7.0/index.html
安装
 pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cu92/torch1.6.0/index.html
安装
 pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cu92/torch1.5.0/index.html
cpu
安装
 pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cpu/torch1.11.0/index.html
安装
 pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cpu/torch1.10.0/index.html
安装
 pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cpu/torch1.9.0/index.html
安装
 pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cpu/torch1.8.0/index.html
安装
 pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cpu/torch1.7.0/index.html
安装
 pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cpu/torch1.6.0/index.html
安装
 pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cpu/torch1.5.0/index.html
- -**注意**:以上提供的预编译包并不囊括所有的 mmcv-full 版本,你可以点击对应链接查看支持的版本。例如,点击 [cu102-torch1.8.0](https://download.openmmlab.com/mmcv/dist/cu102/torch1.8.0/index.html),可以看到 `cu102-torch1.8.0` 只提供了 1.3.0 及以上的 mmcv-full 版本。另外,从 `mmcv v1.3.17` 开始,我们不再提供`PyTorch 1.3 & 1.4` 对应的 mmcv-full 预编译包。你可以在 [这](./docs/zh_cn/get_started/previous_versions.md) 找到 `PyTorch 1.3 & 1.4` 对应的预编包。虽然我们不再提供 `PyTorch 1.3 & 1.4` 对应的预编译包,但是我们依然在 CI 中保证对它们的兼容持续到下一年。 - -**注意**:mmcv-full 没有提供 Windows 平台 `cu102-torch1.8.0` 和 `cu92-torch*` 的预编译包。 - -除了使用预编译包之外,另一种方式是在本地进行编译,直接运行下述命令 - -```python -pip install mmcv-full -``` +更多安装方式请参考[安装文档](https://mmcv.readthedocs.io/zh_CN/latest/get_started/installation.html)。 -但注意本地编译可能会耗时 10 分钟以上。 +### 安装 mmcv -b. 安装精简版 +如果你需要使用和 PyTorch 相关的模块,请确保 PyTorch 已经成功安装在环境中,可以参考 [PyTorch 官方安装文档](https://github.com/pytorch/pytorch#installation)。 -```python -pip install mmcv +```bash +pip install -U openmim +mim install mmcv ``` -c. 安装完整版并且编译 onnxruntime 的自定义算子 - -- 详细的指南请查看[这里](docs/zh_cn/deployment/onnxruntime_op.md)。 +## 支持的开源项目 -如果想从源码编译 MMCV,请参考[该文档](https://mmcv.readthedocs.io/zh_CN/latest/get_started/build.html)。 +- [MIM](https://github.com/open-mmlab/mim): MIM 是 OpenMMlab 项目、算法、模型的统一入口 +- [MMClassification](https://github.com/open-mmlab/mmclassification): OpenMMLab 图像分类工具箱 +- [MMDetection](https://github.com/open-mmlab/mmdetection): OpenMMLab 目标检测工具箱 +- [MMDetection3D](https://github.com/open-mmlab/mmdetection3d): OpenMMLab 新一代通用 3D 目标检测平台 +- [MMRotate](https://github.com/open-mmlab/mmrotate): OpenMMLab 旋转框检测工具箱与测试基准 +- [MMSegmentation](https://github.com/open-mmlab/mmsegmentation): OpenMMLab 语义分割工具箱 +- [MMOCR](https://github.com/open-mmlab/mmocr): OpenMMLab 全流程文字检测识别理解工具箱 +- [MMPose](https://github.com/open-mmlab/mmpose): OpenMMLab 姿态估计工具箱 +- [MMHuman3D](https://github.com/open-mmlab/mmhuman3d): OpenMMLab 人体参数化模型工具箱与测试基准 +- [MMSelfSup](https://github.com/open-mmlab/mmselfsup): OpenMMLab 自监督学习工具箱与测试基准 +- [MMRazor](https://github.com/open-mmlab/mmrazor): OpenMMLab 模型压缩工具箱与测试基准 +- [MMFewShot](https://github.com/open-mmlab/mmfewshot): OpenMMLab 少样本学习工具箱与测试基准 +- [MMAction2](https://github.com/open-mmlab/mmaction2): OpenMMLab 新一代视频理解工具箱 +- [MMTracking](https://github.com/open-mmlab/mmtracking): OpenMMLab 一体化视频目标感知平台 +- [MMFlow](https://github.com/open-mmlab/mmflow): OpenMMLab 光流估计工具箱与测试基准 +- [MMEditing](https://github.com/open-mmlab/mmediting): OpenMMLab 图像视频编辑工具箱 +- [MMGeneration](https://github.com/open-mmlab/mmgeneration): OpenMMLab 图片视频生成模型工具箱 +- [MMDeploy](https://github.com/open-mmlab/mmdeploy): OpenMMLab 模型部署框架 ## FAQ -如果你遇到了安装问题,CUDA 相关的问题或者 RuntimeErrors,可以首先参考[问题解决页面](https://mmcv.readthedocs.io/zh_CN/latest/faq.html) 看是否已经有解决方案。 +如果你遇到了安装问题或者运行时问题,请查看[问题解决页面](https://mmcv.readthedocs.io/zh_CN/latest/faq.html)是否已有解决方案。如果问题仍然没有解决,欢迎提 [issue](https://github.com/open-mmlab/mmcv/issues)。 ## 贡献指南 @@ -254,11 +139,11 @@ c. 安装完整版并且编译 onnxruntime 的自定义算子 ## 许可证 -`MMCV` 目前以 Apache 2.0 的许可证发布,但是其中有一部分功能并不是使用的 Apache2.0 许可证,我们在 [许可证](LICENSES.md) 中详细地列出了这些功能以及他们对应的许可证,如果您正在从事盈利性活动,请谨慎参考此文档。 +`MMCV` 目前以 Apache 2.0 的许可证发布,但是其中有一部分功能并不是使用的 Apache2.0 许可证,我们在[许可证](LICENSES.md)中详细地列出了这些功能以及他们对应的许可证,如果您正在从事盈利性活动,请谨慎参考此文档。 ## 欢迎加入 OpenMMLab 社区 -扫描下方的二维码可关注 OpenMMLab 团队的 [知乎官方账号](https://www.zhihu.com/people/openmmlab),加入 OpenMMLab 团队的 [官方交流 QQ 群](https://jq.qq.com/?_wv=1027&k=3ijNTqfg),或添加微信小助手”OpenMMLabwx“加入官方交流微信群。 +扫描下方的二维码可关注 OpenMMLab 团队的[知乎官方账号](https://www.zhihu.com/people/openmmlab),加入 OpenMMLab 团队的[官方交流 QQ 群](https://jq.qq.com/?_wv=1027&k=3ijNTqfg),或添加微信小助手”OpenMMLabwx“加入官方交流微信群。
diff --git a/docker/release/Dockerfile b/docker/release/Dockerfile index 493aa6d1625..1f10acc2bf6 100644 --- a/docker/release/Dockerfile +++ b/docker/release/Dockerfile @@ -16,5 +16,8 @@ RUN apt-get update && apt-get install -y libgl1 libglib2.0-0 \ && rm -rf /var/lib/apt/lists/* # Install mmcv-full -ARG MMCV="1.5.1" -RUN pip install openmim && mim install mmcv-full==${MMCV} && python -c 'import mmcv;print(mmcv.__version__)' +ARG MMCV="" +RUN if [ "${MMCV}" = "" ]; then pip install -U openmim && mim install mmcv-full; else pip install -U openmim && mim install mmcv-full==${MMCV}; fi + +# Verify the installation +RUN python -c 'import mmcv;print(mmcv.__version__)' diff --git a/docs/en/_static/version.json b/docs/en/_static/version.json new file mode 100644 index 00000000000..a1643eedbb8 --- /dev/null +++ b/docs/en/_static/version.json @@ -0,0 +1,1931 @@ +{ + "Linux": [ + { + "cuda": "11.6", + "torch": "1.12.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0" + ] + }, + { + "cuda": "11.5", + "torch": "1.11.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0", + "1.5.3", + "1.5.2", + "1.5.1", + "1.5.0", + "1.4.8", + "1.4.7" + ] + }, + { + "cuda": "11.3", + "torch": "1.12.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0" + ] + }, + { + "cuda": "11.3", + "torch": "1.11.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0", + "1.5.3", + "1.5.2", + "1.5.1", + "1.5.0", + "1.4.8", + "1.4.7" + ] + }, + { + "cuda": "11.3", + "torch": "1.10.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0", + "1.5.3", + "1.5.2", + "1.5.1", + "1.5.0", + "1.4.8", + "1.4.7", + "1.4.6", + "1.4.5", + "1.4.4", + "1.4.3", + "1.4.2", + "1.4.1", + "1.4.0", + "1.3.18", + "1.3.17", + "1.3.16" + ] + }, + { + "cuda": "11.1", + "torch": "1.10.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0", + "1.5.3", + "1.5.2", + "1.5.1", + "1.5.0", + "1.4.8", + "1.4.7", + "1.4.6", + "1.4.5", + "1.4.4", + "1.4.3", + "1.4.2", + "1.4.1", + "1.4.0", + "1.3.18", + "1.3.17", + "1.3.16" + ] + }, + { + "cuda": "11.1", + "torch": "1.9.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0", + "1.5.3", + "1.5.2", + "1.5.1", + "1.5.0", + "1.4.8", + "1.4.7", + "1.4.6", + "1.4.5", + "1.4.4", + "1.4.3", + "1.4.2", + "1.4.1", + "1.4.0", + "1.3.9", + "1.3.18", + "1.3.17", + "1.3.16", + "1.3.15", + "1.3.14", + "1.3.13", + "1.3.12", + "1.3.11", + "1.3.10" + ] + }, + { + "cuda": "11.1", + "torch": "1.8.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0", + "1.5.3", + "1.5.2", + "1.5.1", + "1.5.0", + "1.4.8", + "1.4.7", + "1.4.6", + "1.4.5", + "1.4.4", + "1.4.3", + "1.4.2", + "1.4.1", + "1.4.0", + "1.3.9", + "1.3.8", + "1.3.7", + "1.3.6", + "1.3.5", + "1.3.4", + "1.3.3", + "1.3.2", + "1.3.18", + "1.3.17", + "1.3.16", + "1.3.15", + "1.3.14", + "1.3.13", + "1.3.12", + "1.3.11", + "1.3.10", + "1.3.1", + "1.3.0" + ] + }, + { + "cuda": "11.0", + "torch": "1.7.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0", + "1.5.3", + "1.5.2", + "1.5.1", + "1.5.0", + "1.4.8", + "1.4.7", + "1.4.6", + "1.4.5", + "1.4.4", + "1.4.3", + "1.4.2", + "1.4.1", + "1.4.0", + "1.3.9", + "1.3.8", + "1.3.7", + "1.3.6", + "1.3.5", + "1.3.4", + "1.3.3", + "1.3.2", + "1.3.18", + "1.3.17", + "1.3.16", + "1.3.15", + "1.3.14", + "1.3.13", + "1.3.12", + "1.3.11", + "1.3.10", + "1.3.1", + "1.3.0", + "1.2.7", + "1.2.6", + "1.2.5", + "1.2.4", + "1.2.3", + "1.2.2", + "1.2.0" + ] + }, + { + "cuda": "10.2", + "torch": "1.12.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0" + ] + }, + { + "cuda": "10.2", + "torch": "1.11.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0", + "1.5.3", + "1.5.2", + "1.5.1", + "1.5.0", + "1.4.8", + "1.4.7" + ] + }, + { + "cuda": "10.2", + "torch": "1.10.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0", + "1.5.3", + "1.5.2", + "1.5.1", + "1.5.0", + "1.4.8", + "1.4.7", + "1.4.6", + "1.4.5", + "1.4.4", + "1.4.3", + "1.4.2", + "1.4.1", + "1.4.0", + "1.3.18", + "1.3.17", + "1.3.16" + ] + }, + { + "cuda": "10.2", + "torch": "1.9.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0", + "1.5.3", + "1.5.2", + "1.5.1", + "1.5.0", + "1.4.8", + "1.4.7", + "1.4.6", + "1.4.5", + "1.4.4", + "1.4.3", + "1.4.2", + "1.4.1", + "1.4.0", + "1.3.9", + "1.3.18", + "1.3.17", + "1.3.16", + "1.3.15", + "1.3.14", + "1.3.13", + "1.3.12", + "1.3.11", + "1.3.10" + ] + }, + { + "cuda": "10.2", + "torch": "1.8.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0", + "1.5.3", + "1.5.2", + "1.5.1", + "1.5.0", + "1.4.8", + "1.4.7", + "1.4.6", + "1.4.5", + "1.4.4", + "1.4.3", + "1.4.2", + "1.4.1", + "1.4.0", + "1.3.9", + "1.3.8", + "1.3.7", + "1.3.6", + "1.3.5", + "1.3.4", + "1.3.3", + "1.3.2", + "1.3.18", + "1.3.17", + "1.3.16", + "1.3.15", + "1.3.14", + "1.3.13", + "1.3.12", + "1.3.11", + "1.3.10", + "1.3.1", + "1.3.0" + ] + }, + { + "cuda": "10.2", + "torch": "1.7.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0", + "1.5.3", + "1.5.2", + "1.5.1", + "1.5.0", + "1.4.8", + "1.4.7", + "1.4.6", + "1.4.5", + "1.4.4", + "1.4.3", + "1.4.2", + "1.4.1", + "1.4.0", + "1.3.9", + "1.3.8", + "1.3.7", + "1.3.6", + "1.3.5", + "1.3.4", + "1.3.3", + "1.3.2", + "1.3.18", + "1.3.17", + "1.3.16", + "1.3.15", + "1.3.14", + "1.3.13", + "1.3.12", + "1.3.11", + "1.3.10", + "1.3.1", + "1.3.0", + "1.2.7", + "1.2.6", + "1.2.5", + "1.2.4", + "1.2.3", + "1.2.2", + "1.2.1", + "1.2.0" + ] + }, + { + "cuda": "10.2", + "torch": "1.6.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0", + "1.5.3", + "1.5.2", + "1.5.1", + "1.5.0", + "1.4.8", + "1.4.7", + "1.4.6", + "1.4.5", + "1.4.4", + "1.4.3", + "1.4.2", + "1.4.1", + "1.4.0", + "1.3.9", + "1.3.8", + "1.3.7", + "1.3.6", + "1.3.5", + "1.3.4", + "1.3.3", + "1.3.2", + "1.3.18", + "1.3.17", + "1.3.16", + "1.3.15", + "1.3.14", + "1.3.13", + "1.3.12", + "1.3.11", + "1.3.10", + "1.3.1", + "1.3.0", + "1.2.7", + "1.2.6", + "1.2.5", + "1.2.4", + "1.2.3", + "1.2.2", + "1.2.1", + "1.2.0", + "1.1.6", + "1.1.5", + "1.1.4", + "1.1.3", + "1.1.2", + "1.1.1", + "1.1.0", + "1.0.5" + ] + }, + { + "cuda": "10.2", + "torch": "1.5.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0", + "1.5.3", + "1.5.2", + "1.5.1", + "1.5.0", + "1.4.8", + "1.4.7", + "1.4.6", + "1.4.5", + "1.4.4", + "1.4.3", + "1.4.2", + "1.4.1", + "1.4.0", + "1.3.9", + "1.3.8", + "1.3.7", + "1.3.6", + "1.3.5", + "1.3.4", + "1.3.3", + "1.3.2", + "1.3.18", + "1.3.17", + "1.3.16", + "1.3.15", + "1.3.14", + "1.3.13", + "1.3.12", + "1.3.11", + "1.3.10", + "1.3.1", + "1.3.0", + "1.2.7", + "1.2.6", + "1.2.5", + "1.2.4", + "1.2.3", + "1.2.2", + "1.2.1", + "1.2.0", + "1.1.6", + "1.1.5", + "1.1.4", + "1.1.3", + "1.1.2", + "1.1.1", + "1.1.0", + "1.0.5", + "1.0.4", + "1.0.3", + "1.0.2", + "1.0.0" + ] + }, + { + "cuda": "10.1", + "torch": "1.8.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0", + "1.5.3", + "1.5.2", + "1.5.1", + "1.5.0", + "1.4.8", + "1.4.7", + "1.4.6", + "1.4.5", + "1.4.4", + "1.4.3", + "1.4.2", + "1.4.1", + "1.4.0", + "1.3.9", + "1.3.8", + "1.3.7", + "1.3.6", + "1.3.5", + "1.3.4", + "1.3.3", + "1.3.2", + "1.3.18", + "1.3.17", + "1.3.16", + "1.3.15", + "1.3.14", + "1.3.13", + "1.3.12", + "1.3.11", + "1.3.10", + "1.3.1", + "1.3.0" + ] + }, + { + "cuda": "10.1", + "torch": "1.7.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0", + "1.5.3", + "1.5.2", + "1.5.1", + "1.5.0", + "1.4.8", + "1.4.7", + "1.4.6", + "1.4.5", + "1.4.4", + "1.4.3", + "1.4.2", + "1.4.1", + "1.4.0", + "1.3.9", + "1.3.8", + "1.3.7", + "1.3.6", + "1.3.5", + "1.3.4", + "1.3.3", + "1.3.2", + "1.3.18", + "1.3.17", + "1.3.16", + "1.3.15", + "1.3.14", + "1.3.13", + "1.3.12", + "1.3.11", + "1.3.10", + "1.3.1", + "1.3.0", + "1.2.7", + "1.2.6", + "1.2.5", + "1.2.4", + "1.2.3", + "1.2.2", + "1.2.1", + "1.2.0" + ] + }, + { + "cuda": "10.1", + "torch": "1.6.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0", + "1.5.3", + "1.5.2", + "1.5.1", + "1.5.0", + "1.4.8", + "1.4.7", + "1.4.6", + "1.4.5", + "1.4.4", + "1.4.3", + "1.4.2", + "1.4.1", + "1.4.0", + "1.3.9", + "1.3.8", + "1.3.7", + "1.3.6", + "1.3.5", + "1.3.4", + "1.3.3", + "1.3.2", + "1.3.18", + "1.3.17", + "1.3.16", + "1.3.15", + "1.3.14", + "1.3.13", + "1.3.12", + "1.3.11", + "1.3.10", + "1.3.1", + "1.3.0", + "1.2.7", + "1.2.6", + "1.2.5", + "1.2.4", + "1.2.3", + "1.2.2", + "1.2.1", + "1.2.0", + "1.1.6", + "1.1.5", + "1.1.4", + "1.1.3", + "1.1.2", + "1.1.1", + "1.1.0", + "1.0.5" + ] + }, + { + "cuda": "10.1", + "torch": "1.5.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0", + "1.5.3", + "1.5.2", + "1.5.1", + "1.5.0", + "1.4.8", + "1.4.7", + "1.4.6", + "1.4.5", + "1.4.4", + "1.4.3", + "1.4.2", + "1.4.1", + "1.4.0", + "1.3.9", + "1.3.8", + "1.3.7", + "1.3.6", + "1.3.5", + "1.3.4", + "1.3.3", + "1.3.2", + "1.3.18", + "1.3.17", + "1.3.16", + "1.3.15", + "1.3.14", + "1.3.13", + "1.3.12", + "1.3.11", + "1.3.10", + "1.3.1", + "1.3.0", + "1.2.7", + "1.2.6", + "1.2.5", + "1.2.4", + "1.2.3", + "1.2.2", + "1.2.1", + "1.2.0", + "1.1.6", + "1.1.5", + "1.1.4", + "1.1.3", + "1.1.2", + "1.1.1", + "1.1.0", + "1.0.5", + "1.0.4", + "1.0.3", + "1.0.2", + "1.0.0" + ] + }, + { + "cuda": "10.1", + "torch": "1.4.x", + "mmcv": [ + "1.2.1", + "1.2.0", + "1.1.6", + "1.1.5", + "1.1.4", + "1.1.3", + "1.1.2", + "1.1.1", + "1.1.0", + "1.0.5", + "1.0.4", + "1.0.3", + "1.0.2", + "1.0.0" + ] + }, + { + "cuda": "10.1", + "torch": "1.3.x", + "mmcv": [ + "1.3.9", + "1.3.8", + "1.3.7", + "1.3.6", + "1.3.5", + "1.3.4", + "1.3.3", + "1.3.2", + "1.3.16", + "1.3.15", + "1.3.14", + "1.3.13", + "1.3.12", + "1.3.11", + "1.3.10", + "1.3.1", + "1.3.0", + "1.2.7", + "1.2.6", + "1.2.5", + "1.2.4", + "1.2.3", + "1.2.1", + "1.2.0", + "1.1.6", + "1.1.5", + "1.1.4", + "1.1.3", + "1.1.2", + "1.1.1", + "1.1.0", + "1.0.5", + "1.0.4", + "1.0.3", + "1.0.2", + "1.0.1", + "1.0.0" + ] + }, + { + "cuda": "10.0", + "torch": "1.4.x", + "mmcv": [ + "1.2.1", + "1.2.0", + "1.1.6", + "1.1.5", + "1.1.4", + "1.1.3", + "1.1.2", + "1.1.1", + "1.1.0", + "1.0.4", + "1.0.3", + "1.0.2", + "1.0.0" + ] + }, + { + "cuda": "10.0", + "torch": "1.3.x", + "mmcv": [ + "1.3.9", + "1.3.8", + "1.3.7", + "1.3.6", + "1.3.5", + "1.3.4", + "1.3.3", + "1.3.2", + "1.3.16", + "1.3.15", + "1.3.14", + "1.3.13", + "1.3.12", + "1.3.11", + "1.3.10", + "1.3.1", + "1.3.0", + "1.2.7", + "1.2.6", + "1.2.5", + "1.2.4", + "1.2.3", + "1.2.1", + "1.2.0", + "1.1.6", + "1.1.5", + "1.1.4", + "1.1.3", + "1.1.2", + "1.1.1", + "1.1.0", + "1.0.5", + "1.0.4", + "1.0.3", + "1.0.2", + "1.0.1", + "1.0.0" + ] + }, + { + "cuda": "9.2", + "torch": "1.7.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0", + "1.5.3", + "1.5.2", + "1.5.1", + "1.5.0", + "1.4.8", + "1.4.7", + "1.4.6", + "1.4.5", + "1.4.4", + "1.4.3", + "1.4.2", + "1.4.1", + "1.4.0", + "1.3.9", + "1.3.8", + "1.3.7", + "1.3.6", + "1.3.5", + "1.3.4", + "1.3.3", + "1.3.2", + "1.3.18", + "1.3.17", + "1.3.16", + "1.3.15", + "1.3.14", + "1.3.13", + "1.3.12", + "1.3.11", + "1.3.10", + "1.3.1", + "1.3.0", + "1.2.7", + "1.2.6", + "1.2.5", + "1.2.4", + "1.2.3", + "1.2.2", + "1.2.1", + "1.2.0" + ] + }, + { + "cuda": "9.2", + "torch": "1.6.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0", + "1.5.3", + "1.5.2", + "1.5.1", + "1.5.0", + "1.4.8", + "1.4.7", + "1.4.6", + "1.4.5", + "1.4.4", + "1.4.3", + "1.4.2", + "1.4.1", + "1.4.0", + "1.3.9", + "1.3.8", + "1.3.7", + "1.3.6", + "1.3.5", + "1.3.4", + "1.3.3", + "1.3.2", + "1.3.18", + "1.3.17", + "1.3.16", + "1.3.15", + "1.3.14", + "1.3.13", + "1.3.12", + "1.3.11", + "1.3.10", + "1.3.1", + "1.3.0", + "1.2.7", + "1.2.6", + "1.2.5", + "1.2.4", + "1.2.3", + "1.2.2", + "1.2.1", + "1.2.0", + "1.1.6", + "1.1.5", + "1.1.4", + "1.1.3", + "1.1.2", + "1.1.1", + "1.1.0", + "1.0.5" + ] + }, + { + "cuda": "9.2", + "torch": "1.5.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0", + "1.5.3", + "1.5.2", + "1.5.1", + "1.5.0", + "1.4.8", + "1.4.7", + "1.4.6", + "1.4.5", + "1.4.4", + "1.4.3", + "1.4.2", + "1.4.1", + "1.4.0", + "1.3.9", + "1.3.8", + "1.3.7", + "1.3.6", + "1.3.5", + "1.3.4", + "1.3.3", + "1.3.2", + "1.3.18", + "1.3.17", + "1.3.16", + "1.3.15", + "1.3.14", + "1.3.13", + "1.3.12", + "1.3.11", + "1.3.10", + "1.3.1", + "1.3.0", + "1.2.7", + "1.2.6", + "1.2.5", + "1.2.4", + "1.2.3", + "1.2.2", + "1.2.1", + "1.2.0", + "1.1.6", + "1.1.5", + "1.1.4", + "1.1.3", + "1.1.2", + "1.1.1", + "1.1.0", + "1.0.5", + "1.0.4", + "1.0.3", + "1.0.2", + "1.0.0" + ] + }, + { + "cuda": "9.2", + "torch": "1.4.x", + "mmcv": [ + "1.2.1", + "1.2.0", + "1.1.6", + "1.1.5", + "1.1.4", + "1.1.3", + "1.1.2", + "1.1.1", + "1.1.0", + "1.0.4", + "1.0.3", + "1.0.2", + "1.0.0" + ] + }, + { + "cuda": "9.2", + "torch": "1.3.x", + "mmcv": [ + "1.3.9", + "1.3.8", + "1.3.7", + "1.3.6", + "1.3.5", + "1.3.4", + "1.3.3", + "1.3.2", + "1.3.16", + "1.3.15", + "1.3.14", + "1.3.13", + "1.3.12", + "1.3.11", + "1.3.10", + "1.3.1", + "1.3.0", + "1.2.7", + "1.2.6", + "1.2.5", + "1.2.4", + "1.2.3", + "1.2.1", + "1.2.0", + "1.1.6", + "1.1.5", + "1.1.4", + "1.1.3", + "1.1.2", + "1.1.1", + "1.1.0", + "1.0.5", + "1.0.4", + "1.0.3", + "1.0.2", + "1.0.1", + "1.0.0" + ] + }, + { + "cuda": "cpu", + "torch": "1.12.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0" + ] + }, + { + "cuda": "cpu", + "torch": "1.11.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0", + "1.5.3", + "1.5.2", + "1.5.1", + "1.5.0", + "1.4.8", + "1.4.7" + ] + }, + { + "cuda": "cpu", + "torch": "1.10.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0", + "1.5.3", + "1.5.2", + "1.5.1", + "1.5.0", + "1.4.8", + "1.4.7", + "1.4.6", + "1.4.5", + "1.4.4", + "1.4.3", + "1.4.2", + "1.4.1", + "1.4.0", + "1.3.18", + "1.3.17", + "1.3.16" + ] + }, + { + "cuda": "cpu", + "torch": "1.9.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0", + "1.5.3", + "1.5.2", + "1.5.1", + "1.5.0", + "1.4.8", + "1.4.7", + "1.4.6", + "1.4.5", + "1.4.4", + "1.4.3", + "1.4.2", + "1.4.1", + "1.4.0", + "1.3.9", + "1.3.18", + "1.3.17", + "1.3.16", + "1.3.15", + "1.3.14", + "1.3.13", + "1.3.12", + "1.3.11", + "1.3.10" + ] + }, + { + "cuda": "cpu", + "torch": "1.8.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0", + "1.5.3", + "1.5.2", + "1.5.1", + "1.5.0", + "1.4.8", + "1.4.7", + "1.4.6", + "1.4.5", + "1.4.4", + "1.4.3", + "1.4.2", + "1.4.1", + "1.4.0", + "1.3.9", + "1.3.8", + "1.3.7", + "1.3.6", + "1.3.5", + "1.3.4", + "1.3.3", + "1.3.2", + "1.3.18", + "1.3.17", + "1.3.16", + "1.3.15", + "1.3.14", + "1.3.13", + "1.3.12", + "1.3.11", + "1.3.10", + "1.3.1", + "1.3.0" + ] + }, + { + "cuda": "cpu", + "torch": "1.7.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0", + "1.5.3", + "1.5.2", + "1.5.1", + "1.5.0", + "1.4.8", + "1.4.7", + "1.4.6", + "1.4.5", + "1.4.4", + "1.4.3", + "1.4.2", + "1.4.1", + "1.4.0", + "1.3.9", + "1.3.8", + "1.3.7", + "1.3.6", + "1.3.5", + "1.3.4", + "1.3.3", + "1.3.2", + "1.3.18", + "1.3.17", + "1.3.16", + "1.3.15", + "1.3.14", + "1.3.13", + "1.3.12", + "1.3.11", + "1.3.10", + "1.3.1", + "1.3.0", + "1.2.7", + "1.2.6", + "1.2.5", + "1.2.4", + "1.2.3", + "1.2.2", + "1.2.1", + "1.2.0" + ] + }, + { + "cuda": "cpu", + "torch": "1.6.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0", + "1.5.3", + "1.5.2", + "1.5.1", + "1.5.0", + "1.4.8", + "1.4.7", + "1.4.6", + "1.4.5", + "1.4.4", + "1.4.3", + "1.4.2", + "1.4.1", + "1.4.0", + "1.3.9", + "1.3.8", + "1.3.7", + "1.3.6", + "1.3.5", + "1.3.4", + "1.3.3", + "1.3.2", + "1.3.18", + "1.3.17", + "1.3.16", + "1.3.15", + "1.3.14", + "1.3.13", + "1.3.12", + "1.3.11", + "1.3.10", + "1.3.1", + "1.3.0", + "1.2.7", + "1.2.6", + "1.2.5", + "1.2.4", + "1.2.3", + "1.2.2", + "1.2.1", + "1.2.0", + "1.1.6", + "1.1.5", + "1.1.4", + "1.1.3", + "1.1.2", + "1.1.1", + "1.1.0", + "1.0.5" + ] + }, + { + "cuda": "cpu", + "torch": "1.5.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0", + "1.5.3", + "1.5.2", + "1.5.1", + "1.5.0", + "1.4.8", + "1.4.7", + "1.4.6", + "1.4.5", + "1.4.4", + "1.4.3", + "1.4.2", + "1.4.1", + "1.4.0", + "1.3.9", + "1.3.8", + "1.3.7", + "1.3.6", + "1.3.5", + "1.3.4", + "1.3.3", + "1.3.2", + "1.3.18", + "1.3.17", + "1.3.16", + "1.3.15", + "1.3.14", + "1.3.13", + "1.3.12", + "1.3.11", + "1.3.10", + "1.3.1", + "1.3.0", + "1.2.7", + "1.2.6", + "1.2.5", + "1.2.4", + "1.2.3", + "1.2.2", + "1.2.1", + "1.2.0", + "1.1.6", + "1.1.5", + "1.1.4", + "1.1.3", + "1.1.2", + "1.1.1", + "1.1.0", + "1.0.5", + "1.0.4", + "1.0.3", + "1.0.2", + "1.0.0" + ] + }, + { + "cuda": "cpu", + "torch": "1.4.x", + "mmcv": [ + "1.2.1", + "1.2.0", + "1.1.6", + "1.1.5", + "1.1.4", + "1.1.3", + "1.1.2", + "1.1.1", + "1.1.0", + "1.0.5", + "1.0.4", + "1.0.3", + "1.0.2", + "1.0.0" + ] + }, + { + "cuda": "cpu", + "torch": "1.3.x", + "mmcv": [ + "1.3.9", + "1.3.8", + "1.3.7", + "1.3.6", + "1.3.5", + "1.3.4", + "1.3.3", + "1.3.2", + "1.3.16", + "1.3.15", + "1.3.14", + "1.3.13", + "1.3.12", + "1.3.11", + "1.3.10", + "1.3.1", + "1.3.0", + "1.2.7", + "1.2.6", + "1.2.5", + "1.2.4", + "1.2.3", + "1.2.1", + "1.2.0", + "1.1.6", + "1.1.5", + "1.1.4", + "1.1.3", + "1.1.2", + "1.1.1", + "1.1.0", + "1.0.5", + "1.0.4", + "1.0.3", + "1.0.2", + "1.0.1", + "1.0.0" + ] + } + ], + "Windows": [ + { + "cuda": "11.6", + "torch": "1.12.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0" + ] + }, + { + "cuda": "11.5", + "torch": "1.11.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0", + "1.5.3", + "1.5.2", + "1.5.1", + "1.5.0", + "1.4.8", + "1.4.7" + ] + }, + { + "cuda": "11.3", + "torch": "1.12.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0" + ] + }, + { + "cuda": "11.3", + "torch": "1.11.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0", + "1.5.3", + "1.5.2", + "1.5.1", + "1.5.0", + "1.4.8", + "1.4.7" + ] + }, + { + "cuda": "11.3", + "torch": "1.10.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0", + "1.5.3", + "1.5.2", + "1.5.1", + "1.5.0", + "1.4.8", + "1.4.7", + "1.4.6", + "1.4.5", + "1.4.4", + "1.4.3", + "1.4.2", + "1.4.1", + "1.4.0" + ] + }, + { + "cuda": "11.1", + "torch": "1.10.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0", + "1.5.3", + "1.5.2", + "1.5.1", + "1.5.0", + "1.4.8", + "1.4.7", + "1.4.6", + "1.4.5", + "1.4.4", + "1.4.3", + "1.4.2", + "1.4.1", + "1.4.0" + ] + }, + { + "cuda": "11.1", + "torch": "1.9.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0", + "1.5.3", + "1.5.2", + "1.5.1", + "1.5.0", + "1.4.8", + "1.4.7", + "1.4.6", + "1.4.5", + "1.4.4", + "1.4.3", + "1.4.2", + "1.4.1", + "1.4.0" + ] + }, + { + "cuda": "11.1", + "torch": "1.8.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0", + "1.5.3", + "1.5.2", + "1.5.1", + "1.5.0", + "1.4.8", + "1.4.7", + "1.4.6", + "1.4.5", + "1.4.4", + "1.4.3", + "1.4.2", + "1.4.1", + "1.4.0" + ] + }, + { + "cuda": "10.2", + "torch": "1.10.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0", + "1.5.3", + "1.5.2", + "1.5.1", + "1.5.0", + "1.4.8", + "1.4.7", + "1.4.6", + "1.4.5", + "1.4.4", + "1.4.3", + "1.4.2", + "1.4.1", + "1.4.0" + ] + }, + { + "cuda": "10.2", + "torch": "1.9.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0", + "1.5.3", + "1.5.2", + "1.5.1", + "1.5.0", + "1.4.8", + "1.4.7", + "1.4.6", + "1.4.5", + "1.4.4", + "1.4.3", + "1.4.2", + "1.4.1", + "1.4.0" + ] + }, + { + "cuda": "10.2", + "torch": "1.8.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0", + "1.5.3", + "1.5.2", + "1.5.1", + "1.5.0", + "1.4.8", + "1.4.7", + "1.4.6", + "1.4.5", + "1.4.4", + "1.4.3", + "1.4.2", + "1.4.1", + "1.4.0" + ] + }, + { + "cuda": "10.2", + "torch": "1.7.x", + "mmcv": [ + "1.4.5", + "1.4.4", + "1.4.3", + "1.4.2", + "1.4.1", + "1.4.0" + ] + }, + { + "cuda": "10.2", + "torch": "1.6.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0", + "1.5.3", + "1.5.2", + "1.5.1", + "1.5.0", + "1.4.8", + "1.4.7", + "1.4.6", + "1.4.5", + "1.4.4", + "1.4.3", + "1.4.2", + "1.4.1", + "1.4.0", + "1.1.5", + "1.1.3" + ] + }, + { + "cuda": "10.2", + "torch": "1.5.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0", + "1.5.3", + "1.5.2", + "1.5.1", + "1.5.0", + "1.4.8", + "1.4.6", + "1.4.5", + "1.4.4", + "1.4.3", + "1.4.2", + "1.4.1", + "1.4.0" + ] + }, + { + "cuda": "10.1", + "torch": "1.8.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0", + "1.5.3", + "1.5.2", + "1.5.1", + "1.5.0", + "1.4.8", + "1.4.6", + "1.4.5", + "1.4.4", + "1.4.3", + "1.4.2", + "1.4.1", + "1.4.0" + ] + }, + { + "cuda": "10.1", + "torch": "1.7.x", + "mmcv": [ + "1.4.5", + "1.4.4", + "1.4.3", + "1.4.2", + "1.4.1", + "1.4.0" + ] + }, + { + "cuda": "10.1", + "torch": "1.6.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0", + "1.5.3", + "1.5.2", + "1.5.1", + "1.5.0", + "1.4.8", + "1.4.7", + "1.4.6", + "1.4.5", + "1.4.4", + "1.4.3", + "1.4.2", + "1.4.1", + "1.4.0", + "1.1.5", + "1.1.3" + ] + }, + { + "cuda": "10.1", + "torch": "1.5.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0", + "1.5.3", + "1.5.2", + "1.5.1", + "1.5.0", + "1.4.8", + "1.4.7", + "1.4.6", + "1.4.5", + "1.4.4", + "1.4.3", + "1.4.2", + "1.4.1", + "1.4.0" + ] + }, + { + "cuda": "cpu", + "torch": "1.12.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0" + ] + }, + { + "cuda": "cpu", + "torch": "1.11.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0", + "1.5.3", + "1.5.2", + "1.5.1", + "1.5.0", + "1.4.8", + "1.4.7" + ] + }, + { + "cuda": "cpu", + "torch": "1.10.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0", + "1.5.3", + "1.5.2", + "1.5.1", + "1.5.0", + "1.4.8", + "1.4.7", + "1.4.6", + "1.4.5", + "1.4.4", + "1.4.3", + "1.4.2", + "1.4.1", + "1.4.0" + ] + }, + { + "cuda": "cpu", + "torch": "1.9.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0", + "1.5.3", + "1.5.2", + "1.5.1", + "1.5.0", + "1.4.8", + "1.4.7", + "1.4.6", + "1.4.5", + "1.4.4", + "1.4.3", + "1.4.2", + "1.4.1", + "1.4.0" + ] + }, + { + "cuda": "cpu", + "torch": "1.8.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0", + "1.5.3", + "1.5.2", + "1.5.1", + "1.5.0", + "1.4.8", + "1.4.7", + "1.4.6", + "1.4.5", + "1.4.4", + "1.4.3", + "1.4.2", + "1.4.1", + "1.4.0" + ] + }, + { + "cuda": "cpu", + "torch": "1.7.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0", + "1.5.3", + "1.5.2", + "1.5.1", + "1.5.0", + "1.4.8", + "1.4.7", + "1.4.6", + "1.4.5", + "1.4.4", + "1.4.3", + "1.4.2", + "1.4.1", + "1.4.0" + ] + }, + { + "cuda": "cpu", + "torch": "1.6.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0", + "1.5.3", + "1.5.2", + "1.5.1", + "1.5.0", + "1.4.8", + "1.4.7", + "1.4.6", + "1.4.5", + "1.4.4", + "1.4.3", + "1.4.2", + "1.4.1", + "1.4.0", + "1.1.5", + "1.1.3" + ] + }, + { + "cuda": "cpu", + "torch": "1.5.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0", + "1.5.3", + "1.5.2", + "1.5.1", + "1.5.0", + "1.4.8", + "1.4.7", + "1.4.6", + "1.4.5", + "1.4.4", + "1.4.3", + "1.4.2", + "1.4.1", + "1.4.0" + ] + }, + { + "cuda": "cpu", + "torch": "1.4.x", + "mmcv": [ + "1.1.5" + ] + } + ] +} diff --git a/docs/en/conf.py b/docs/en/conf.py index e38dfab1d26..08f5f7eb7df 100644 --- a/docs/en/conf.py +++ b/docs/en/conf.py @@ -56,6 +56,8 @@ myst_heading_anchors = 4 +myst_enable_extensions = ['colon_fence'] + autodoc_mock_imports = ['mmcv._ext', 'mmcv.utils.ext_loader', 'torchvision'] # Add any paths that contain templates here, relative to this directory. diff --git a/docs/en/get_started/build.md b/docs/en/get_started/build.md index d987c1e17e2..c03bea59125 100644 --- a/docs/en/get_started/build.md +++ b/docs/en/get_started/build.md @@ -1,58 +1,124 @@ ## Build MMCV from source -### Build on Linux or macOS +### Build mmcv-full -After cloning the repo with +Before installing mmcv-full, make sure that PyTorch has been successfully installed following the [PyTorch official installation guide](https://pytorch.org/get-started/locally/#start-locally). This can be verified using the following command ```bash -git clone https://github.com/open-mmlab/mmcv.git -cd mmcv +python -c 'import torch;print(torch.__version__)' ``` -It is recommended to install `ninja` to speed up the compilation +If version information is output, then PyTorch is installed. -```bash -pip install -r requirements/optional.txt +```{note} +- To compile ONNX Runtime custom operators, please refer to [How to build custom operators for ONNX Runtime](https://mmcv.readthedocs.io/en/latest/deployment/onnxruntime_op.html#how-to-build-custom-operators-for-onnx-runtime) +- To compile TensorRT customization, please refer to [How to build TensorRT plugins in MMCV](https://mmcv.readthedocs.io/en/latest/deployment/tensorrt_plugin.html#how-to-build-tensorrt-plugins-in-mmcv) ``` -You can either +```{note} +If you would like to use `opencv-python-headless` instead of `opencv-python`, +e.g., in a minimum container environment or servers without GUI, +you can first install it before installing MMCV to skip the installation of `opencv-python`. +``` -- install the lite version +#### Build on Linux - ```bash - pip install -e . - ``` +1. Clone the repo -- install the full version + ```bash + git clone https://github.com/open-mmlab/mmcv.git + cd mmcv + ``` - ```bash - MMCV_WITH_OPS=1 pip install -e . - ``` +2. Install `ninja` and `psutil` to speed up the compilation -If you are on macOS, add the following environment variables before the installing command. + ```bash + pip install -r requirements/optional.txt + ``` -```bash -CC=clang CXX=clang++ CFLAGS='-stdlib=libc++' -``` +3. Check the nvcc version (requires 9.2+. Skip if no GPU available.) -e.g., + ```bash + nvcc --version + ``` -```bash -CC=clang CXX=clang++ CFLAGS='-stdlib=libc++' MMCV_WITH_OPS=1 pip install -e . -``` + If the above command outputs the following message, it means that the nvcc setting is OK, otherwise you need to set CUDA_HOME. + + ``` + nvcc: NVIDIA (R) Cuda compiler driver + Copyright (c) 2005-2020 NVIDIA Corporation + Built on Mon_Nov_30_19:08:53_PST_2020 + Cuda compilation tools, release 11.2, V11.2.67 + Build cuda_11.2.r11.2/compiler.29373293_0 + ``` + + :::{note} + If you want to support ROCm, you can refer to [AMD ROCm](https://rocmdocs.amd.com/en/latest/Installation_Guide/Installation-Guide.html) to install ROCm. + ::: + +4. Check the gcc version (requires 5.4+) + + ```bash + gcc --version + ``` + +5. Start building (takes 10+ min) + + ```bash + MMCV_WITH_OPS=1 pip install -e . -v + ``` + +6. Validate the installation + + ```bash + python .dev_scripts/check_installation.py + ``` + + If no error is reported by the above command, the installation is successful. If there is an error reported, please check [Frequently Asked Questions](../faq.md) to see if there is already a solution. + + If no solution is found, please feel free to open an [issue](https://github.com/open-mmlab/mmcv/issues). + +#### Build on macOS ```{note} -If you would like to use `opencv-python-headless` instead of `opencv-python`, -e.g., in a minimum container environment or servers without GUI, -you can first install it before installing MMCV to skip the installation of `opencv-python`. +If you are using a mac with an M1 chip, install the nightly version of PyTorch, otherwise you will encounter the problem in [issues#2218](https://github.com/open-mmlab/mmcv/issues/2218). ``` -### Build on Windows +1. Clone the repo + + ```bash + git clone https://github.com/open-mmlab/mmcv.git + cd mmcv + ``` + +2. Install `ninja` and `psutil` to speed up the compilation + + ```bash + pip install -r requirements/optional.txt + ``` + +3. Start building + + ```bash + MMCV_WITH_OPS=1 pip install -e . + ``` + +4. Validate the installation + + ```bash + python .dev_scripts/check_installation.py + ``` + + If no error is reported by the above command, the installation is successful. If there is an error reported, please check [Frequently Asked Questions](../faq.md) to see if there is already a solution. + + If no solution is found, please feel free to open an [issue](https://github.com/open-mmlab/mmcv/issues). + +#### Build on Windows Building MMCV on Windows is a bit more complicated than that on Linux. The following instructions show how to get this accomplished. -#### Prerequisite +##### Prerequisite The following software is required for building MMCV on windows. Install them first. @@ -71,7 +137,7 @@ Install them first. You should know how to set up environment variables, especially `Path`, on Windows. The following instruction relies heavily on this skill. ``` -#### Setup Python Environment +##### Common steps 1. Launch Anaconda prompt from Windows Start menu @@ -79,62 +145,45 @@ You should know how to set up environment variables, especially `Path`, on Windo 2. Create a new conda environment - ```shell - conda create --name mmcv python=3.7 # 3.6, 3.7, 3.8 should work too as tested - conda activate mmcv # make sure to activate environment before any operation + ```powershell + (base) PS C:\Users\xxx> conda create --name mmcv python=3.7 + (base) PS C:\Users\xxx> conda activate mmcv # make sure to activate environment before any operation ``` 3. Install PyTorch. Choose a version based on your need. - ```shell - conda install pytorch torchvision cudatoolkit=10.2 -c pytorch + ```powershell + # CUDA version + (mmcv) PS C:\Users\xxx> conda install pytorch torchvision cudatoolkit=10.2 -c pytorch + # CPU version + (mmcv) PS C:\Users\xxx> conda install install pytorch torchvision cpuonly -c pytorch ``` - We only tested PyTorch version >= 1.6.0. - -4. Prepare MMCV source code +4. Clone the repo - ```shell - git clone https://github.com/open-mmlab/mmcv.git - cd mmcv + ```powershell + (mmcv) PS C:\Users\xxx> git clone https://github.com/open-mmlab/mmcv.git + (mmcv) PS C:\Users\xxx\mmcv> cd mmcv ``` -5. Install required Python packages +5. Install `ninja` and `psutil` to speed up the compilation - ```shell - pip3 install -r requirements/runtime.txt + ```powershell + (mmcv) PS C:\Users\xxx\mmcv> pip install -r requirements/optional.txt ``` -6. It is recommended to install `ninja` to speed up the compilation +6. Install required Python packages - ```bash - pip install -r requirements/optional.txt + ```shell + (mmcv) PS C:\Users\xxx\mmcv> pip install -r requirements/runtime.txt ``` -#### Build and install MMCV - -MMCV can be built in three ways: - -1. Lite version (without ops) - - In this way, no custom ops are compiled and mmcv is a pure python package. - -2. Full version (CPU ops) - - Module `ops` will be compiled as a pytorch extension, but only x86 code will be compiled. The compiled ops can be executed on CPU only. - -3. Full version (CUDA ops) - - Both x86 and CUDA codes of `ops` module will be compiled. The compiled version can be run on both CPU and CUDA-enabled GPU (if implemented). - -##### Common steps - -1. Set up MSVC compiler +7. Set up MSVC compiler Set Environment variable, add `C:\Program Files (x86)\Microsoft Visual Studio\2019\Community\VC\Tools\MSVC\14.27.29110\bin\Hostx86\x64` to `PATH`, so that `cl.exe` will be available in prompt, as shown below. - ```none - (base) PS C:\Users\xxx> cl + ```powershell + (mmcv) PS C:\Users\xxx\mmcv> cl Microsoft (R) C/C++ Optimizing Compiler Version 19.27.29111 for x64 Copyright (C) Microsoft Corporation. All rights reserved. @@ -145,124 +194,135 @@ MMCV can be built in three ways: You may want to change the system language to English because pytorch will parse text output from `cl.exe` to check its version. However only utf-8 is recognized. Navigate to Control Panel -> Region -> Administrative -> Language for Non-Unicode programs and change it to English. -##### Option 1: Build MMCV (lite version) +##### Build and install MMCV -After finishing above common steps, launch Anaconda shell from Start menu and issue the following commands: +mmcv-full can be built in two ways: -```shell -# activate environment -conda activate mmcv -# change directory -cd mmcv -# install -python setup.py develop -# check -pip list -``` +1. Full version (CPU ops) -##### Option 2: Build MMCV (full version with CPU) + Module `ops` will be compiled as a pytorch extension, but only x86 code will be compiled. The compiled ops can be executed on CPU only. -1. Finish above common steps +2. Full version (CUDA ops) -2. Set up environment variables + Both x86 and CUDA codes of `ops` module will be compiled. The compiled version can be run on both CPU and CUDA-enabled GPU (if implemented). - ```shell - $env:MMCV_WITH_OPS = 1 - $env:MAX_JOBS = 8 # based on your available number of CPU cores and amount of memory - ``` +###### CPU version -3. Following build steps of the lite version +1. Set up environment variables - ```shell - # activate environment - conda activate mmcv - # change directory - cd mmcv - # build - python setup.py build_ext # if success, cl will be launched to compile ops - # install - python setup.py develop - # check - pip list + ```powershell + (mmcv) PS C:\Users\xxx\mmcv> $env:MMCV_WITH_OPS = 1 ``` -##### Option 3: Build MMCV (full version with CUDA) +2. Build and install -1. Finish above common steps + ```powershell + (mmcv) PS C:\Users\xxx\mmcv> python setup.py build_ext + (mmcv) PS C:\Users\xxx\mmcv> python setup.py develop + ``` + +###### GPU version 2. Make sure `CUDA_PATH` or `CUDA_HOME` is already set in `envs` via `ls env:`, desired output is shown as below: - ```none - (base) PS C:\Users\WRH> ls env: + ```powershell + (mmcv) PS C:\Users\xxx\mmcv> ls env: Name Value ---- ----- - <... omit some lines ...> CUDA_PATH C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2 CUDA_PATH_V10_1 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1 CUDA_PATH_V10_2 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2 - <... omit some lines ...> ``` This should already be done by CUDA installer. If not, or you have multiple version of CUDA toolkit installed, set it with - ```shell - $env:CUDA_HOME = "C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2" + ```powershell + (mmcv) PS C:\Users\xxx\mmcv> $env:CUDA_HOME = "C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2" # OR - $env:CUDA_HOME = $env:CUDA_PATH_V10_2 # if CUDA_PATH_V10_2 is in envs: + (mmcv) PS C:\Users\xxx\mmcv> $env:CUDA_HOME = $env:CUDA_PATH_V10_2 # if CUDA_PATH_V10_2 is in envs: ``` 3. Set CUDA target arch ```shell - # Suppose you are using GTX 1080, which is of capability 6.1 - $env:TORCH_CUDA_ARCH_LIST="6.1" - # OR build all supported arch, will be slow - $env:TORCH_CUDA_ARCH_LIST="3.5 3.7 5.0 5.2 6.0 6.1 7.0 7.5" + # Here you need to change to the target architecture corresponding to your GPU + (mmcv) PS C:\Users\xxx\mmcv> $env:TORCH_CUDA_ARCH_LIST="7.5" ``` -```{note} -Check your the compute capability of your GPU from [here](https://developer.nvidia.com/cuda-gpus). -``` + :::{note} + Check your the compute capability of your GPU from [here](https://developer.nvidia.com/cuda-gpus). -4. Launch compiling the same way as CPU + ```powershell + (mmcv) PS C:\Users\xxx\mmcv> &"C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\extras\demo_suite\deviceQuery.exe" + Device 0: "NVIDIA GeForce GTX 1660 SUPER" + CUDA Driver Version / Runtime Version 11.7 / 11.1 + CUDA Capability Major/Minor version number: 7.5 + ``` - ```shell - $env:MMCV_WITH_OPS = 1 - $env:MAX_JOBS = 8 # based on available number of CPU cores and amount of memory - # activate environment - conda activate mmcv - # change directory - cd mmcv + The 7.5 above indicates the target architecture. Note: You need to replace v10.2 with your CUDA version in the above command. + ::: + +4. Build and install + + ```powershell # build python setup.py build_ext # if success, cl will be launched to compile ops # install python setup.py develop - # check - pip list ``` -```{note} -If you are compiling against PyTorch 1.6.0, you might meet some errors from PyTorch as described in [this issue](https://github.com/pytorch/pytorch/issues/42467). Follow [this pull request](https://github.com/pytorch/pytorch/pull/43380/files) to modify the source code in your local PyTorch installation. + ```{note} + If you are compiling against PyTorch 1.6.0, you might meet some errors from PyTorch as described in [this issue](https://github.com/pytorch/pytorch/issues/42467). Follow [this pull request](https://github.com/pytorch/pytorch/pull/43380/files) to modify the source code in your local PyTorch installation. + ``` + +##### Validate installation + +```powershell +(mmcv) PS C:\Users\xxx\mmcv> python .dev_scripts/check_installation.py ``` -If you meet issues when running or compiling mmcv, we list some common issues in [Frequently Asked Question](../faq.html). +If no error is reported by the above command, the installation is successful. If there is an error reported, please check [Frequently Asked Questions](../faq.md) to see if there is already a solution. +If no solution is found, please feel free to open an [issue](https://github.com/open-mmlab/mmcv/issues). + +### Build mmcv + +If you need to use PyTorch-related modules, make sure PyTorch has been successfully installed in your environment by referring to the [PyTorch official installation guide](https://github.com/pytorch/pytorch#installation). + +1. Clone the repo -## \[Optional\] Build MMCV on IPU machine + ```bash + git clone https://github.com/open-mmlab/mmcv.git + cd mmcv + ``` + +2. Start building + + ```bash + pip install -e . -v + ``` + +3. Validate installation + + ```bash + python -c 'import mmcv;print(mmcv.__version__)' + ``` + +### Build mmcv-full on IPU machine Firstly, you need to apply for an IPU cloud machine, see [here](https://www.graphcore.ai/ipus-in-the-cloud). -### Option 1: Docker +#### Option 1: Docker 1. Pull docker -```shell - docker pull graphcore/pytorch -``` + ```bash + docker pull graphcore/pytorch + ``` 2. Build MMCV under same python environment -### Option 2: Install from SDK +#### Option 2: Install from SDK 1. Build MMCV diff --git a/docs/en/get_started/installation.md b/docs/en/get_started/installation.md index d9fd1b33607..ea1a095aae5 100644 --- a/docs/en/get_started/installation.md +++ b/docs/en/get_started/installation.md @@ -2,183 +2,374 @@ There are two versions of MMCV: -- **mmcv-full**: comprehensive, with full features and various CUDA ops out of box. It takes longer time to build. -- **mmcv**: lite, without CUDA ops but all other features, similar to mmcv\<1.0.0. It is useful when you do not need those CUDA ops. +- **mmcv-full**: comprehensive, with full features and various CPU and CUDA ops out of box. It takes longer time to build. +- **mmcv**: lite, without CPU and CUDA ops but all other features, similar to mmcv\<1.0.0. It is useful when you do not need those CUDA ops. ```{warning} Do not install both versions in the same environment, otherwise you may encounter errors like `ModuleNotFound`. You need to uninstall one before installing the other. `Installing the full version is highly recommended if CUDA is avaliable`. ``` -a. Install the full version. +### Install mmcv-full -Before installing mmcv-full, make sure that PyTorch has been successfully installed following the [official guide](https://pytorch.org/). +```{note} +- The following installation steps are for Linux and Windows platforms only. To install mmcv-full on macOS platform, please refer to [build mmcv-full from source](https://mmcv.readthedocs.io/en/latest/get_started/build.html#macos-mmcv-full). +- To compile ONNX Runtime custom operators, please refer to [How to build custom operators for ONNX Runtime](https://mmcv.readthedocs.io/en/latest/deployment/onnxruntime_op.html#how-to-build-custom-operators-for-onnx-runtime) +- To compile TensorRT customization, please refer to [How to build TensorRT plugins in MMCV](https://mmcv.readthedocs.io/en/latest/deployment/tensorrt_plugin.html#how-to-build-tensorrt-plugins-in-mmcv) +``` -We provide pre-built mmcv packages (recommended) with different PyTorch and CUDA versions to simplify the building for **Linux and Windows systems**. In addition, you can run [check_installation.py](.dev_scripts/check_installation.py) to check the installation of mmcv-full after running the installation commands. +Before installing mmcv-full, make sure that PyTorch has been successfully installed following the [PyTorch official installation guide](https://pytorch.org/get-started/locally/#start-locally). This can be verified using the following command -i. Install the latest version. +```bash +python -c 'import torch;print(torch.__version__)' +``` -The rule for installing the latest `mmcv-full` is as follows: +If version information is output, then PyTorch is installed. -```shell -pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/{cu_version}/{torch_version}/index.html -``` +#### Install with mim (recommended) -Please replace `{cu_version}` and `{torch_version}` in the url to your desired one. For example, -to install the latest `mmcv-full` with `CUDA 11.1` and `PyTorch 1.9.0`, use the following command: +[mim](https://github.com/open-mmlab/mim) is the package management tool for the OpenMMLab projects, which makes it easy to install mmcv-full -```shell -pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/cu111/torch1.9.0/index.html +```bash +pip install -U openmim +mim install mmcv-full ``` -For more details, please refer the the following tables and delete `=={mmcv_version}`. +If you find that the above installation command does not use a pre-built package ending with `.whl` but a source package ending with `.tar.gz`, you may not have a pre-build package corresponding to the PyTorch or CUDA or mmcv-full version, in which case you can [build mmcv-full from source](build.md). + +
+Installation log using pre-built packages + +Looking in links: https://download.openmmlab.com/mmcv/dist/cu102/torch1.8.0/index.html
+Collecting mmcv-full
+Downloading https://download.openmmlab.com/mmcv/dist/cu102/torch1.8.0/mmcv_full-1.6.1-cp38-cp38-manylinux1_x86_64.whl + +
+ +
+Installation log using source packages + +Looking in links: https://download.openmmlab.com/mmcv/dist/cu102/torch1.8.0/index.html
+Collecting mmcv-full==1.6.0
+Downloading mmcv-full-1.6.0.tar.gz -ii. Install a specified version. +
-The rule for installing a specified `mmcv-full` is as follows: +To install a specific version of mmcv-full, for example, mmcv-full version 1.6.0, you can use the following command -```shell -pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/{cu_version}/{torch_version}/index.html +```bash +mim install mmcv-full==1.6.0 ``` -First of all, please refer to the Releases and replace `{mmcv_version}` a specified one. e.g. `1.3.9`. -Then replace `{cu_version}` and `{torch_version}` in the url to your desired versions. For example, -to install `mmcv-full==1.3.9` with `CUDA 11.1` and `PyTorch 1.9.0`, use the following command: +:::{note} +If you would like to use `opencv-python-headless` instead of `opencv-python`, +e.g., in a minimum container environment or servers without GUI, +you can first install it before installing MMCV to skip the installation of `opencv-python`. -```shell -pip install mmcv-full==1.3.9 -f https://download.openmmlab.com/mmcv/dist/cu111/torch1.9.0/index.html +Alternatively, if it takes too long to install a dependency library, you can specify the pypi source + +```bash +mim install mmcv-full -i https://pypi.tuna.tsinghua.edu.cn/simple ``` -```{note} +::: + +You can run [check_installation.py](https://github.com/open-mmlab/mmcv/.dev_scripts/check_installation.py) to check the installation of mmcv-full after running the installation commands. + +#### Install with pip + +Use the following command to check the version of CUDA and PyTorch + +```bash +python -c 'import torch;print(torch.__version__);print(torch.version.cuda)' +``` + +Select the appropriate installation command depending on the type of system, CUDA version, PyTorch version, and MMCV version + + + + +
+ + + + +
+

+
+
+
+
+If you do not find a corresponding version in the dropdown box above, you probably do not have a pre-built package corresponding to the PyTorch or CUDA or mmcv-full version, at which point you can [build mmcv-full from source](build.md).
+
+:::{note}
 mmcv-full is only compiled on PyTorch 1.x.0 because the compatibility
 usually holds between 1.x.0 and 1.x.1. If your PyTorch version is 1.x.1, you
 can install mmcv-full compiled with PyTorch 1.x.0 and it usually works well.
-For example, if your PyTorch version is 1.8.1 and CUDA version is 11.1, you
-can use the following command to install mmcv-full.
+For example, if your PyTorch version is 1.8.1, you can feel free to choose 1.8.x.
+:::
+
+:::{note}
+If you would like to use `opencv-python-headless` instead of `opencv-python`,
+e.g., in a minimum container environment or servers without GUI,
+you can first install it before installing MMCV to skip the installation of `opencv-python`.
 
-`pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/cu111/torch1.8.0/index.html`
+Alternatively, if it takes too long to install a dependency library, you can specify the pypi source
+
+```bash
+mim install mmcv-full -i https://pypi.tuna.tsinghua.edu.cn/simple
 ```
 
-For more details, please refer the the following tables.
-
-
-  
-    
-      
-      
-      
-      
-      
-      
-      
-      
-    
-    
-      
-      
-      
-      
-      
-      
-      
-      
-    
-    
-      
-      
-      
-      
-      
-      
-      
-      
-    
-    
-      
-      
-      
-      
-      
-      
-      
-      
-    
-    
-      
-      
-      
-      
-      
-      
-      
-      
-    
-    
-      
-      
-      
-      
-      
-      
-      
-      
-    
-    
-      
-      
-      
-      
-      
-      
-      
-      
-    
-    
-      
-      
-      
-      
-      
-      
-      
-      
-    
-    
-      
-      
-      
-      
-      
-      
-      
-      
-    
-  
-
CUDA torch 1.11torch 1.10torch 1.9torch 1.8torch 1.7torch 1.6torch 1.5
11.5
install
pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cu115/torch1.11.0/index.html
11.3
install
pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cu113/torch1.11.0/index.html
install
pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cu113/torch1.10.0/index.html
11.1
install
pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cu111/torch1.10.0/index.html
install
pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cu111/torch1.9.0/index.html
install
pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cu111/torch1.8.0/index.html
11.0
install
pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cu110/torch1.7.0/index.html
10.2
install
pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cu102/torch1.11.0/index.html
install
pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cu102/torch1.10.0/index.html
install
pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cu102/torch1.9.0/index.html
install
pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cu102/torch1.8.0/index.html
install
pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cu102/torch1.7.0/index.html
install
pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cu102/torch1.6.0/index.html
install
pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cu102/torch1.5.0/index.html
10.1
install
 pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cu101/torch1.8.0/index.html
install
 pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cu101/torch1.7.0/index.html
install
 pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cu101/torch1.6.0/index.html
install
 pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cu101/torch1.5.0/index.html
9.2
install
 pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cu92/torch1.7.0/index.html
install
 pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cu92/torch1.6.0/index.html
install
 pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cu92/torch1.5.0/index.html
cpu
install
 pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cpu/torch1.11.0/index.html
install
 pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cpu/torch1.10.0/index.html
install
 pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cpu/torch1.9.0/index.html
install
 pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cpu/torch1.8.0/index.html
install
 pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cpu/torch1.7.0/index.html
install
 pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cpu/torch1.6.0/index.html
install
 pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cpu/torch1.5.0/index.html
+::: -```{note} -The pre-built packages provided above do not include all versions of mmcv-full, you can click on the corresponding links to see the supported versions. For example, if you click [cu102-torch1.8.0](https://download.openmmlab.com/mmcv/dist/cu102/torch1.8.0/index.html), you can see that `cu102-torch1.8.0` only provides 1.3.0 and above versions of mmcv-full. In addition, We no longer provide `mmcv-full` pre-built packages compiled with `PyTorch 1.3 & 1.4` since v1.3.17. You can find previous versions that compiled with PyTorch 1.3 & 1.4 [here](./previous_versions.md). The compatibility is still ensured in our CI, but we will discard the support of PyTorch 1.3 & 1.4 next year. +You can run [check_installation.py](https://github.com/open-mmlab/mmcv/.dev_scripts/check_installation.py) to check the installation of mmcv-full after running the installation commands. + +#### Using mmcv-full with Docker + +Build with local repository + +```bash +git clone https://github.com/open-mmlab/mmcv.git && cd mmcv +docker build -t mmcv -f docker/release/Dockerfile . ``` -```{note} -mmcv-full does not provide pre-built packages for `cu102-torch1.11` and `cu92-torch*` on Windows. +Or build with remote repository + +```bash +docker build -t mmcv https://github.com/open-mmlab/mmcv.git#master:docker/release ``` -Another way is to compile locally by running +The [Dockerfile](release/Dockerfile) installs latest released version of mmcv-full by default, but you can specify mmcv versions to install expected versions. -```python -pip install mmcv-full +```bash +docker image build -t mmcv -f docker/release/Dockerfile --build-arg MMCV=1.5.0 . ``` -Note that the local compiling may take up to 10 mins. +If you also want to use other versions of PyTorch and CUDA, you can also pass them when building docker images. -b. Install the lite version. +An example to build an image with PyTorch 1.11 and CUDA 11.3. -```python -pip install mmcv +```bash +docker build -t mmcv -f docker/release/Dockerfile \ + --build-arg PYTORCH=1.9.0 \ + --build-arg CUDA=11.1 \ + --build-arg CUDNN=8 \ + --build-arg MMCV=1.5.0 . ``` -c. Install full version with custom operators for onnxruntime +More available versions of PyTorch and CUDA can be found at [dockerhub/pytorch](https://hub.docker.com/r/pytorch/pytorch/tags). -- Check [here](https://mmcv.readthedocs.io/en/latest/deployment/onnxruntime_custom_ops.html) for detailed instruction. +### Install mmcv -If you would like to build MMCV from source, please refer to the [guide](https://mmcv.readthedocs.io/en/latest/get_started/build.html). +If you need to use PyTorch-related modules, make sure PyTorch has been successfully installed in your environment by referring to the [PyTorch official installation guide](https://github.com/pytorch/pytorch#installation). + +```python +pip install mmcv +``` diff --git a/docs/en/get_started/introduction.md b/docs/en/get_started/introduction.md index 9ef6ee99dc4..711d731c4d8 100644 --- a/docs/en/get_started/introduction.md +++ b/docs/en/get_started/introduction.md @@ -22,17 +22,17 @@ research projects as below: - [MMGeneration](https://github.com/open-mmlab/mmgeneration): OpenMMLab image and video generative models toolbox. - [MMDeploy](https://github.com/open-mmlab/mmdeploy): OpenMMLab model deployment framework. -It provides the following functionalities. +It provides the following functionalities: -- Universal IO APIs -- Image/Video processing -- Image and annotation visualization -- Useful utilities (progress bar, timer, ...) -- PyTorch runner with hooking mechanism -- Various CNN architectures -- High-quality implementation of common CUDA ops +- [Universal IO APIs](https://mmcv.readthedocs.io/en/latest/understand_mmcv/io.html) +- [Image/Video processing](https://mmcv.readthedocs.io/en/latest/understand_mmcv/data_process.html) +- [Image and annotation visualization](https://mmcv.readthedocs.io/en/latest/understand_mmcv/visualization.html) +- [Useful utilities (progress bar, timer, ...)](https://mmcv.readthedocs.io/en/latest/understand_mmcv/utils.html) +- [PyTorch runner with hooking mechanism](https://mmcv.readthedocs.io/en/latest/understand_mmcv/runner.html) +- [Various CNN architectures](https://mmcv.readthedocs.io/en/latest/understand_mmcv/cnn.html) +- [High-quality implementation of common CPU and CUDA ops](https://mmcv.readthedocs.io/en/latest/understand_mmcv/ops.html) -It supports the following systems. +It supports the following systems: - Linux - Windows diff --git a/docs/zh_cn/_static/version.json b/docs/zh_cn/_static/version.json new file mode 100644 index 00000000000..a1643eedbb8 --- /dev/null +++ b/docs/zh_cn/_static/version.json @@ -0,0 +1,1931 @@ +{ + "Linux": [ + { + "cuda": "11.6", + "torch": "1.12.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0" + ] + }, + { + "cuda": "11.5", + "torch": "1.11.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0", + "1.5.3", + "1.5.2", + "1.5.1", + "1.5.0", + "1.4.8", + "1.4.7" + ] + }, + { + "cuda": "11.3", + "torch": "1.12.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0" + ] + }, + { + "cuda": "11.3", + "torch": "1.11.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0", + "1.5.3", + "1.5.2", + "1.5.1", + "1.5.0", + "1.4.8", + "1.4.7" + ] + }, + { + "cuda": "11.3", + "torch": "1.10.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0", + "1.5.3", + "1.5.2", + "1.5.1", + "1.5.0", + "1.4.8", + "1.4.7", + "1.4.6", + "1.4.5", + "1.4.4", + "1.4.3", + "1.4.2", + "1.4.1", + "1.4.0", + "1.3.18", + "1.3.17", + "1.3.16" + ] + }, + { + "cuda": "11.1", + "torch": "1.10.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0", + "1.5.3", + "1.5.2", + "1.5.1", + "1.5.0", + "1.4.8", + "1.4.7", + "1.4.6", + "1.4.5", + "1.4.4", + "1.4.3", + "1.4.2", + "1.4.1", + "1.4.0", + "1.3.18", + "1.3.17", + "1.3.16" + ] + }, + { + "cuda": "11.1", + "torch": "1.9.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0", + "1.5.3", + "1.5.2", + "1.5.1", + "1.5.0", + "1.4.8", + "1.4.7", + "1.4.6", + "1.4.5", + "1.4.4", + "1.4.3", + "1.4.2", + "1.4.1", + "1.4.0", + "1.3.9", + "1.3.18", + "1.3.17", + "1.3.16", + "1.3.15", + "1.3.14", + "1.3.13", + "1.3.12", + "1.3.11", + "1.3.10" + ] + }, + { + "cuda": "11.1", + "torch": "1.8.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0", + "1.5.3", + "1.5.2", + "1.5.1", + "1.5.0", + "1.4.8", + "1.4.7", + "1.4.6", + "1.4.5", + "1.4.4", + "1.4.3", + "1.4.2", + "1.4.1", + "1.4.0", + "1.3.9", + "1.3.8", + "1.3.7", + "1.3.6", + "1.3.5", + "1.3.4", + "1.3.3", + "1.3.2", + "1.3.18", + "1.3.17", + "1.3.16", + "1.3.15", + "1.3.14", + "1.3.13", + "1.3.12", + "1.3.11", + "1.3.10", + "1.3.1", + "1.3.0" + ] + }, + { + "cuda": "11.0", + "torch": "1.7.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0", + "1.5.3", + "1.5.2", + "1.5.1", + "1.5.0", + "1.4.8", + "1.4.7", + "1.4.6", + "1.4.5", + "1.4.4", + "1.4.3", + "1.4.2", + "1.4.1", + "1.4.0", + "1.3.9", + "1.3.8", + "1.3.7", + "1.3.6", + "1.3.5", + "1.3.4", + "1.3.3", + "1.3.2", + "1.3.18", + "1.3.17", + "1.3.16", + "1.3.15", + "1.3.14", + "1.3.13", + "1.3.12", + "1.3.11", + "1.3.10", + "1.3.1", + "1.3.0", + "1.2.7", + "1.2.6", + "1.2.5", + "1.2.4", + "1.2.3", + "1.2.2", + "1.2.0" + ] + }, + { + "cuda": "10.2", + "torch": "1.12.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0" + ] + }, + { + "cuda": "10.2", + "torch": "1.11.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0", + "1.5.3", + "1.5.2", + "1.5.1", + "1.5.0", + "1.4.8", + "1.4.7" + ] + }, + { + "cuda": "10.2", + "torch": "1.10.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0", + "1.5.3", + "1.5.2", + "1.5.1", + "1.5.0", + "1.4.8", + "1.4.7", + "1.4.6", + "1.4.5", + "1.4.4", + "1.4.3", + "1.4.2", + "1.4.1", + "1.4.0", + "1.3.18", + "1.3.17", + "1.3.16" + ] + }, + { + "cuda": "10.2", + "torch": "1.9.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0", + "1.5.3", + "1.5.2", + "1.5.1", + "1.5.0", + "1.4.8", + "1.4.7", + "1.4.6", + "1.4.5", + "1.4.4", + "1.4.3", + "1.4.2", + "1.4.1", + "1.4.0", + "1.3.9", + "1.3.18", + "1.3.17", + "1.3.16", + "1.3.15", + "1.3.14", + "1.3.13", + "1.3.12", + "1.3.11", + "1.3.10" + ] + }, + { + "cuda": "10.2", + "torch": "1.8.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0", + "1.5.3", + "1.5.2", + "1.5.1", + "1.5.0", + "1.4.8", + "1.4.7", + "1.4.6", + "1.4.5", + "1.4.4", + "1.4.3", + "1.4.2", + "1.4.1", + "1.4.0", + "1.3.9", + "1.3.8", + "1.3.7", + "1.3.6", + "1.3.5", + "1.3.4", + "1.3.3", + "1.3.2", + "1.3.18", + "1.3.17", + "1.3.16", + "1.3.15", + "1.3.14", + "1.3.13", + "1.3.12", + "1.3.11", + "1.3.10", + "1.3.1", + "1.3.0" + ] + }, + { + "cuda": "10.2", + "torch": "1.7.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0", + "1.5.3", + "1.5.2", + "1.5.1", + "1.5.0", + "1.4.8", + "1.4.7", + "1.4.6", + "1.4.5", + "1.4.4", + "1.4.3", + "1.4.2", + "1.4.1", + "1.4.0", + "1.3.9", + "1.3.8", + "1.3.7", + "1.3.6", + "1.3.5", + "1.3.4", + "1.3.3", + "1.3.2", + "1.3.18", + "1.3.17", + "1.3.16", + "1.3.15", + "1.3.14", + "1.3.13", + "1.3.12", + "1.3.11", + "1.3.10", + "1.3.1", + "1.3.0", + "1.2.7", + "1.2.6", + "1.2.5", + "1.2.4", + "1.2.3", + "1.2.2", + "1.2.1", + "1.2.0" + ] + }, + { + "cuda": "10.2", + "torch": "1.6.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0", + "1.5.3", + "1.5.2", + "1.5.1", + "1.5.0", + "1.4.8", + "1.4.7", + "1.4.6", + "1.4.5", + "1.4.4", + "1.4.3", + "1.4.2", + "1.4.1", + "1.4.0", + "1.3.9", + "1.3.8", + "1.3.7", + "1.3.6", + "1.3.5", + "1.3.4", + "1.3.3", + "1.3.2", + "1.3.18", + "1.3.17", + "1.3.16", + "1.3.15", + "1.3.14", + "1.3.13", + "1.3.12", + "1.3.11", + "1.3.10", + "1.3.1", + "1.3.0", + "1.2.7", + "1.2.6", + "1.2.5", + "1.2.4", + "1.2.3", + "1.2.2", + "1.2.1", + "1.2.0", + "1.1.6", + "1.1.5", + "1.1.4", + "1.1.3", + "1.1.2", + "1.1.1", + "1.1.0", + "1.0.5" + ] + }, + { + "cuda": "10.2", + "torch": "1.5.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0", + "1.5.3", + "1.5.2", + "1.5.1", + "1.5.0", + "1.4.8", + "1.4.7", + "1.4.6", + "1.4.5", + "1.4.4", + "1.4.3", + "1.4.2", + "1.4.1", + "1.4.0", + "1.3.9", + "1.3.8", + "1.3.7", + "1.3.6", + "1.3.5", + "1.3.4", + "1.3.3", + "1.3.2", + "1.3.18", + "1.3.17", + "1.3.16", + "1.3.15", + "1.3.14", + "1.3.13", + "1.3.12", + "1.3.11", + "1.3.10", + "1.3.1", + "1.3.0", + "1.2.7", + "1.2.6", + "1.2.5", + "1.2.4", + "1.2.3", + "1.2.2", + "1.2.1", + "1.2.0", + "1.1.6", + "1.1.5", + "1.1.4", + "1.1.3", + "1.1.2", + "1.1.1", + "1.1.0", + "1.0.5", + "1.0.4", + "1.0.3", + "1.0.2", + "1.0.0" + ] + }, + { + "cuda": "10.1", + "torch": "1.8.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0", + "1.5.3", + "1.5.2", + "1.5.1", + "1.5.0", + "1.4.8", + "1.4.7", + "1.4.6", + "1.4.5", + "1.4.4", + "1.4.3", + "1.4.2", + "1.4.1", + "1.4.0", + "1.3.9", + "1.3.8", + "1.3.7", + "1.3.6", + "1.3.5", + "1.3.4", + "1.3.3", + "1.3.2", + "1.3.18", + "1.3.17", + "1.3.16", + "1.3.15", + "1.3.14", + "1.3.13", + "1.3.12", + "1.3.11", + "1.3.10", + "1.3.1", + "1.3.0" + ] + }, + { + "cuda": "10.1", + "torch": "1.7.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0", + "1.5.3", + "1.5.2", + "1.5.1", + "1.5.0", + "1.4.8", + "1.4.7", + "1.4.6", + "1.4.5", + "1.4.4", + "1.4.3", + "1.4.2", + "1.4.1", + "1.4.0", + "1.3.9", + "1.3.8", + "1.3.7", + "1.3.6", + "1.3.5", + "1.3.4", + "1.3.3", + "1.3.2", + "1.3.18", + "1.3.17", + "1.3.16", + "1.3.15", + "1.3.14", + "1.3.13", + "1.3.12", + "1.3.11", + "1.3.10", + "1.3.1", + "1.3.0", + "1.2.7", + "1.2.6", + "1.2.5", + "1.2.4", + "1.2.3", + "1.2.2", + "1.2.1", + "1.2.0" + ] + }, + { + "cuda": "10.1", + "torch": "1.6.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0", + "1.5.3", + "1.5.2", + "1.5.1", + "1.5.0", + "1.4.8", + "1.4.7", + "1.4.6", + "1.4.5", + "1.4.4", + "1.4.3", + "1.4.2", + "1.4.1", + "1.4.0", + "1.3.9", + "1.3.8", + "1.3.7", + "1.3.6", + "1.3.5", + "1.3.4", + "1.3.3", + "1.3.2", + "1.3.18", + "1.3.17", + "1.3.16", + "1.3.15", + "1.3.14", + "1.3.13", + "1.3.12", + "1.3.11", + "1.3.10", + "1.3.1", + "1.3.0", + "1.2.7", + "1.2.6", + "1.2.5", + "1.2.4", + "1.2.3", + "1.2.2", + "1.2.1", + "1.2.0", + "1.1.6", + "1.1.5", + "1.1.4", + "1.1.3", + "1.1.2", + "1.1.1", + "1.1.0", + "1.0.5" + ] + }, + { + "cuda": "10.1", + "torch": "1.5.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0", + "1.5.3", + "1.5.2", + "1.5.1", + "1.5.0", + "1.4.8", + "1.4.7", + "1.4.6", + "1.4.5", + "1.4.4", + "1.4.3", + "1.4.2", + "1.4.1", + "1.4.0", + "1.3.9", + "1.3.8", + "1.3.7", + "1.3.6", + "1.3.5", + "1.3.4", + "1.3.3", + "1.3.2", + "1.3.18", + "1.3.17", + "1.3.16", + "1.3.15", + "1.3.14", + "1.3.13", + "1.3.12", + "1.3.11", + "1.3.10", + "1.3.1", + "1.3.0", + "1.2.7", + "1.2.6", + "1.2.5", + "1.2.4", + "1.2.3", + "1.2.2", + "1.2.1", + "1.2.0", + "1.1.6", + "1.1.5", + "1.1.4", + "1.1.3", + "1.1.2", + "1.1.1", + "1.1.0", + "1.0.5", + "1.0.4", + "1.0.3", + "1.0.2", + "1.0.0" + ] + }, + { + "cuda": "10.1", + "torch": "1.4.x", + "mmcv": [ + "1.2.1", + "1.2.0", + "1.1.6", + "1.1.5", + "1.1.4", + "1.1.3", + "1.1.2", + "1.1.1", + "1.1.0", + "1.0.5", + "1.0.4", + "1.0.3", + "1.0.2", + "1.0.0" + ] + }, + { + "cuda": "10.1", + "torch": "1.3.x", + "mmcv": [ + "1.3.9", + "1.3.8", + "1.3.7", + "1.3.6", + "1.3.5", + "1.3.4", + "1.3.3", + "1.3.2", + "1.3.16", + "1.3.15", + "1.3.14", + "1.3.13", + "1.3.12", + "1.3.11", + "1.3.10", + "1.3.1", + "1.3.0", + "1.2.7", + "1.2.6", + "1.2.5", + "1.2.4", + "1.2.3", + "1.2.1", + "1.2.0", + "1.1.6", + "1.1.5", + "1.1.4", + "1.1.3", + "1.1.2", + "1.1.1", + "1.1.0", + "1.0.5", + "1.0.4", + "1.0.3", + "1.0.2", + "1.0.1", + "1.0.0" + ] + }, + { + "cuda": "10.0", + "torch": "1.4.x", + "mmcv": [ + "1.2.1", + "1.2.0", + "1.1.6", + "1.1.5", + "1.1.4", + "1.1.3", + "1.1.2", + "1.1.1", + "1.1.0", + "1.0.4", + "1.0.3", + "1.0.2", + "1.0.0" + ] + }, + { + "cuda": "10.0", + "torch": "1.3.x", + "mmcv": [ + "1.3.9", + "1.3.8", + "1.3.7", + "1.3.6", + "1.3.5", + "1.3.4", + "1.3.3", + "1.3.2", + "1.3.16", + "1.3.15", + "1.3.14", + "1.3.13", + "1.3.12", + "1.3.11", + "1.3.10", + "1.3.1", + "1.3.0", + "1.2.7", + "1.2.6", + "1.2.5", + "1.2.4", + "1.2.3", + "1.2.1", + "1.2.0", + "1.1.6", + "1.1.5", + "1.1.4", + "1.1.3", + "1.1.2", + "1.1.1", + "1.1.0", + "1.0.5", + "1.0.4", + "1.0.3", + "1.0.2", + "1.0.1", + "1.0.0" + ] + }, + { + "cuda": "9.2", + "torch": "1.7.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0", + "1.5.3", + "1.5.2", + "1.5.1", + "1.5.0", + "1.4.8", + "1.4.7", + "1.4.6", + "1.4.5", + "1.4.4", + "1.4.3", + "1.4.2", + "1.4.1", + "1.4.0", + "1.3.9", + "1.3.8", + "1.3.7", + "1.3.6", + "1.3.5", + "1.3.4", + "1.3.3", + "1.3.2", + "1.3.18", + "1.3.17", + "1.3.16", + "1.3.15", + "1.3.14", + "1.3.13", + "1.3.12", + "1.3.11", + "1.3.10", + "1.3.1", + "1.3.0", + "1.2.7", + "1.2.6", + "1.2.5", + "1.2.4", + "1.2.3", + "1.2.2", + "1.2.1", + "1.2.0" + ] + }, + { + "cuda": "9.2", + "torch": "1.6.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0", + "1.5.3", + "1.5.2", + "1.5.1", + "1.5.0", + "1.4.8", + "1.4.7", + "1.4.6", + "1.4.5", + "1.4.4", + "1.4.3", + "1.4.2", + "1.4.1", + "1.4.0", + "1.3.9", + "1.3.8", + "1.3.7", + "1.3.6", + "1.3.5", + "1.3.4", + "1.3.3", + "1.3.2", + "1.3.18", + "1.3.17", + "1.3.16", + "1.3.15", + "1.3.14", + "1.3.13", + "1.3.12", + "1.3.11", + "1.3.10", + "1.3.1", + "1.3.0", + "1.2.7", + "1.2.6", + "1.2.5", + "1.2.4", + "1.2.3", + "1.2.2", + "1.2.1", + "1.2.0", + "1.1.6", + "1.1.5", + "1.1.4", + "1.1.3", + "1.1.2", + "1.1.1", + "1.1.0", + "1.0.5" + ] + }, + { + "cuda": "9.2", + "torch": "1.5.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0", + "1.5.3", + "1.5.2", + "1.5.1", + "1.5.0", + "1.4.8", + "1.4.7", + "1.4.6", + "1.4.5", + "1.4.4", + "1.4.3", + "1.4.2", + "1.4.1", + "1.4.0", + "1.3.9", + "1.3.8", + "1.3.7", + "1.3.6", + "1.3.5", + "1.3.4", + "1.3.3", + "1.3.2", + "1.3.18", + "1.3.17", + "1.3.16", + "1.3.15", + "1.3.14", + "1.3.13", + "1.3.12", + "1.3.11", + "1.3.10", + "1.3.1", + "1.3.0", + "1.2.7", + "1.2.6", + "1.2.5", + "1.2.4", + "1.2.3", + "1.2.2", + "1.2.1", + "1.2.0", + "1.1.6", + "1.1.5", + "1.1.4", + "1.1.3", + "1.1.2", + "1.1.1", + "1.1.0", + "1.0.5", + "1.0.4", + "1.0.3", + "1.0.2", + "1.0.0" + ] + }, + { + "cuda": "9.2", + "torch": "1.4.x", + "mmcv": [ + "1.2.1", + "1.2.0", + "1.1.6", + "1.1.5", + "1.1.4", + "1.1.3", + "1.1.2", + "1.1.1", + "1.1.0", + "1.0.4", + "1.0.3", + "1.0.2", + "1.0.0" + ] + }, + { + "cuda": "9.2", + "torch": "1.3.x", + "mmcv": [ + "1.3.9", + "1.3.8", + "1.3.7", + "1.3.6", + "1.3.5", + "1.3.4", + "1.3.3", + "1.3.2", + "1.3.16", + "1.3.15", + "1.3.14", + "1.3.13", + "1.3.12", + "1.3.11", + "1.3.10", + "1.3.1", + "1.3.0", + "1.2.7", + "1.2.6", + "1.2.5", + "1.2.4", + "1.2.3", + "1.2.1", + "1.2.0", + "1.1.6", + "1.1.5", + "1.1.4", + "1.1.3", + "1.1.2", + "1.1.1", + "1.1.0", + "1.0.5", + "1.0.4", + "1.0.3", + "1.0.2", + "1.0.1", + "1.0.0" + ] + }, + { + "cuda": "cpu", + "torch": "1.12.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0" + ] + }, + { + "cuda": "cpu", + "torch": "1.11.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0", + "1.5.3", + "1.5.2", + "1.5.1", + "1.5.0", + "1.4.8", + "1.4.7" + ] + }, + { + "cuda": "cpu", + "torch": "1.10.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0", + "1.5.3", + "1.5.2", + "1.5.1", + "1.5.0", + "1.4.8", + "1.4.7", + "1.4.6", + "1.4.5", + "1.4.4", + "1.4.3", + "1.4.2", + "1.4.1", + "1.4.0", + "1.3.18", + "1.3.17", + "1.3.16" + ] + }, + { + "cuda": "cpu", + "torch": "1.9.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0", + "1.5.3", + "1.5.2", + "1.5.1", + "1.5.0", + "1.4.8", + "1.4.7", + "1.4.6", + "1.4.5", + "1.4.4", + "1.4.3", + "1.4.2", + "1.4.1", + "1.4.0", + "1.3.9", + "1.3.18", + "1.3.17", + "1.3.16", + "1.3.15", + "1.3.14", + "1.3.13", + "1.3.12", + "1.3.11", + "1.3.10" + ] + }, + { + "cuda": "cpu", + "torch": "1.8.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0", + "1.5.3", + "1.5.2", + "1.5.1", + "1.5.0", + "1.4.8", + "1.4.7", + "1.4.6", + "1.4.5", + "1.4.4", + "1.4.3", + "1.4.2", + "1.4.1", + "1.4.0", + "1.3.9", + "1.3.8", + "1.3.7", + "1.3.6", + "1.3.5", + "1.3.4", + "1.3.3", + "1.3.2", + "1.3.18", + "1.3.17", + "1.3.16", + "1.3.15", + "1.3.14", + "1.3.13", + "1.3.12", + "1.3.11", + "1.3.10", + "1.3.1", + "1.3.0" + ] + }, + { + "cuda": "cpu", + "torch": "1.7.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0", + "1.5.3", + "1.5.2", + "1.5.1", + "1.5.0", + "1.4.8", + "1.4.7", + "1.4.6", + "1.4.5", + "1.4.4", + "1.4.3", + "1.4.2", + "1.4.1", + "1.4.0", + "1.3.9", + "1.3.8", + "1.3.7", + "1.3.6", + "1.3.5", + "1.3.4", + "1.3.3", + "1.3.2", + "1.3.18", + "1.3.17", + "1.3.16", + "1.3.15", + "1.3.14", + "1.3.13", + "1.3.12", + "1.3.11", + "1.3.10", + "1.3.1", + "1.3.0", + "1.2.7", + "1.2.6", + "1.2.5", + "1.2.4", + "1.2.3", + "1.2.2", + "1.2.1", + "1.2.0" + ] + }, + { + "cuda": "cpu", + "torch": "1.6.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0", + "1.5.3", + "1.5.2", + "1.5.1", + "1.5.0", + "1.4.8", + "1.4.7", + "1.4.6", + "1.4.5", + "1.4.4", + "1.4.3", + "1.4.2", + "1.4.1", + "1.4.0", + "1.3.9", + "1.3.8", + "1.3.7", + "1.3.6", + "1.3.5", + "1.3.4", + "1.3.3", + "1.3.2", + "1.3.18", + "1.3.17", + "1.3.16", + "1.3.15", + "1.3.14", + "1.3.13", + "1.3.12", + "1.3.11", + "1.3.10", + "1.3.1", + "1.3.0", + "1.2.7", + "1.2.6", + "1.2.5", + "1.2.4", + "1.2.3", + "1.2.2", + "1.2.1", + "1.2.0", + "1.1.6", + "1.1.5", + "1.1.4", + "1.1.3", + "1.1.2", + "1.1.1", + "1.1.0", + "1.0.5" + ] + }, + { + "cuda": "cpu", + "torch": "1.5.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0", + "1.5.3", + "1.5.2", + "1.5.1", + "1.5.0", + "1.4.8", + "1.4.7", + "1.4.6", + "1.4.5", + "1.4.4", + "1.4.3", + "1.4.2", + "1.4.1", + "1.4.0", + "1.3.9", + "1.3.8", + "1.3.7", + "1.3.6", + "1.3.5", + "1.3.4", + "1.3.3", + "1.3.2", + "1.3.18", + "1.3.17", + "1.3.16", + "1.3.15", + "1.3.14", + "1.3.13", + "1.3.12", + "1.3.11", + "1.3.10", + "1.3.1", + "1.3.0", + "1.2.7", + "1.2.6", + "1.2.5", + "1.2.4", + "1.2.3", + "1.2.2", + "1.2.1", + "1.2.0", + "1.1.6", + "1.1.5", + "1.1.4", + "1.1.3", + "1.1.2", + "1.1.1", + "1.1.0", + "1.0.5", + "1.0.4", + "1.0.3", + "1.0.2", + "1.0.0" + ] + }, + { + "cuda": "cpu", + "torch": "1.4.x", + "mmcv": [ + "1.2.1", + "1.2.0", + "1.1.6", + "1.1.5", + "1.1.4", + "1.1.3", + "1.1.2", + "1.1.1", + "1.1.0", + "1.0.5", + "1.0.4", + "1.0.3", + "1.0.2", + "1.0.0" + ] + }, + { + "cuda": "cpu", + "torch": "1.3.x", + "mmcv": [ + "1.3.9", + "1.3.8", + "1.3.7", + "1.3.6", + "1.3.5", + "1.3.4", + "1.3.3", + "1.3.2", + "1.3.16", + "1.3.15", + "1.3.14", + "1.3.13", + "1.3.12", + "1.3.11", + "1.3.10", + "1.3.1", + "1.3.0", + "1.2.7", + "1.2.6", + "1.2.5", + "1.2.4", + "1.2.3", + "1.2.1", + "1.2.0", + "1.1.6", + "1.1.5", + "1.1.4", + "1.1.3", + "1.1.2", + "1.1.1", + "1.1.0", + "1.0.5", + "1.0.4", + "1.0.3", + "1.0.2", + "1.0.1", + "1.0.0" + ] + } + ], + "Windows": [ + { + "cuda": "11.6", + "torch": "1.12.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0" + ] + }, + { + "cuda": "11.5", + "torch": "1.11.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0", + "1.5.3", + "1.5.2", + "1.5.1", + "1.5.0", + "1.4.8", + "1.4.7" + ] + }, + { + "cuda": "11.3", + "torch": "1.12.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0" + ] + }, + { + "cuda": "11.3", + "torch": "1.11.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0", + "1.5.3", + "1.5.2", + "1.5.1", + "1.5.0", + "1.4.8", + "1.4.7" + ] + }, + { + "cuda": "11.3", + "torch": "1.10.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0", + "1.5.3", + "1.5.2", + "1.5.1", + "1.5.0", + "1.4.8", + "1.4.7", + "1.4.6", + "1.4.5", + "1.4.4", + "1.4.3", + "1.4.2", + "1.4.1", + "1.4.0" + ] + }, + { + "cuda": "11.1", + "torch": "1.10.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0", + "1.5.3", + "1.5.2", + "1.5.1", + "1.5.0", + "1.4.8", + "1.4.7", + "1.4.6", + "1.4.5", + "1.4.4", + "1.4.3", + "1.4.2", + "1.4.1", + "1.4.0" + ] + }, + { + "cuda": "11.1", + "torch": "1.9.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0", + "1.5.3", + "1.5.2", + "1.5.1", + "1.5.0", + "1.4.8", + "1.4.7", + "1.4.6", + "1.4.5", + "1.4.4", + "1.4.3", + "1.4.2", + "1.4.1", + "1.4.0" + ] + }, + { + "cuda": "11.1", + "torch": "1.8.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0", + "1.5.3", + "1.5.2", + "1.5.1", + "1.5.0", + "1.4.8", + "1.4.7", + "1.4.6", + "1.4.5", + "1.4.4", + "1.4.3", + "1.4.2", + "1.4.1", + "1.4.0" + ] + }, + { + "cuda": "10.2", + "torch": "1.10.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0", + "1.5.3", + "1.5.2", + "1.5.1", + "1.5.0", + "1.4.8", + "1.4.7", + "1.4.6", + "1.4.5", + "1.4.4", + "1.4.3", + "1.4.2", + "1.4.1", + "1.4.0" + ] + }, + { + "cuda": "10.2", + "torch": "1.9.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0", + "1.5.3", + "1.5.2", + "1.5.1", + "1.5.0", + "1.4.8", + "1.4.7", + "1.4.6", + "1.4.5", + "1.4.4", + "1.4.3", + "1.4.2", + "1.4.1", + "1.4.0" + ] + }, + { + "cuda": "10.2", + "torch": "1.8.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0", + "1.5.3", + "1.5.2", + "1.5.1", + "1.5.0", + "1.4.8", + "1.4.7", + "1.4.6", + "1.4.5", + "1.4.4", + "1.4.3", + "1.4.2", + "1.4.1", + "1.4.0" + ] + }, + { + "cuda": "10.2", + "torch": "1.7.x", + "mmcv": [ + "1.4.5", + "1.4.4", + "1.4.3", + "1.4.2", + "1.4.1", + "1.4.0" + ] + }, + { + "cuda": "10.2", + "torch": "1.6.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0", + "1.5.3", + "1.5.2", + "1.5.1", + "1.5.0", + "1.4.8", + "1.4.7", + "1.4.6", + "1.4.5", + "1.4.4", + "1.4.3", + "1.4.2", + "1.4.1", + "1.4.0", + "1.1.5", + "1.1.3" + ] + }, + { + "cuda": "10.2", + "torch": "1.5.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0", + "1.5.3", + "1.5.2", + "1.5.1", + "1.5.0", + "1.4.8", + "1.4.6", + "1.4.5", + "1.4.4", + "1.4.3", + "1.4.2", + "1.4.1", + "1.4.0" + ] + }, + { + "cuda": "10.1", + "torch": "1.8.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0", + "1.5.3", + "1.5.2", + "1.5.1", + "1.5.0", + "1.4.8", + "1.4.6", + "1.4.5", + "1.4.4", + "1.4.3", + "1.4.2", + "1.4.1", + "1.4.0" + ] + }, + { + "cuda": "10.1", + "torch": "1.7.x", + "mmcv": [ + "1.4.5", + "1.4.4", + "1.4.3", + "1.4.2", + "1.4.1", + "1.4.0" + ] + }, + { + "cuda": "10.1", + "torch": "1.6.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0", + "1.5.3", + "1.5.2", + "1.5.1", + "1.5.0", + "1.4.8", + "1.4.7", + "1.4.6", + "1.4.5", + "1.4.4", + "1.4.3", + "1.4.2", + "1.4.1", + "1.4.0", + "1.1.5", + "1.1.3" + ] + }, + { + "cuda": "10.1", + "torch": "1.5.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0", + "1.5.3", + "1.5.2", + "1.5.1", + "1.5.0", + "1.4.8", + "1.4.7", + "1.4.6", + "1.4.5", + "1.4.4", + "1.4.3", + "1.4.2", + "1.4.1", + "1.4.0" + ] + }, + { + "cuda": "cpu", + "torch": "1.12.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0" + ] + }, + { + "cuda": "cpu", + "torch": "1.11.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0", + "1.5.3", + "1.5.2", + "1.5.1", + "1.5.0", + "1.4.8", + "1.4.7" + ] + }, + { + "cuda": "cpu", + "torch": "1.10.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0", + "1.5.3", + "1.5.2", + "1.5.1", + "1.5.0", + "1.4.8", + "1.4.7", + "1.4.6", + "1.4.5", + "1.4.4", + "1.4.3", + "1.4.2", + "1.4.1", + "1.4.0" + ] + }, + { + "cuda": "cpu", + "torch": "1.9.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0", + "1.5.3", + "1.5.2", + "1.5.1", + "1.5.0", + "1.4.8", + "1.4.7", + "1.4.6", + "1.4.5", + "1.4.4", + "1.4.3", + "1.4.2", + "1.4.1", + "1.4.0" + ] + }, + { + "cuda": "cpu", + "torch": "1.8.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0", + "1.5.3", + "1.5.2", + "1.5.1", + "1.5.0", + "1.4.8", + "1.4.7", + "1.4.6", + "1.4.5", + "1.4.4", + "1.4.3", + "1.4.2", + "1.4.1", + "1.4.0" + ] + }, + { + "cuda": "cpu", + "torch": "1.7.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0", + "1.5.3", + "1.5.2", + "1.5.1", + "1.5.0", + "1.4.8", + "1.4.7", + "1.4.6", + "1.4.5", + "1.4.4", + "1.4.3", + "1.4.2", + "1.4.1", + "1.4.0" + ] + }, + { + "cuda": "cpu", + "torch": "1.6.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0", + "1.5.3", + "1.5.2", + "1.5.1", + "1.5.0", + "1.4.8", + "1.4.7", + "1.4.6", + "1.4.5", + "1.4.4", + "1.4.3", + "1.4.2", + "1.4.1", + "1.4.0", + "1.1.5", + "1.1.3" + ] + }, + { + "cuda": "cpu", + "torch": "1.5.x", + "mmcv": [ + "1.6.2", + "1.6.1", + "1.6.0", + "1.5.3", + "1.5.2", + "1.5.1", + "1.5.0", + "1.4.8", + "1.4.7", + "1.4.6", + "1.4.5", + "1.4.4", + "1.4.3", + "1.4.2", + "1.4.1", + "1.4.0" + ] + }, + { + "cuda": "cpu", + "torch": "1.4.x", + "mmcv": [ + "1.1.5" + ] + } + ] +} diff --git a/docs/zh_cn/conf.py b/docs/zh_cn/conf.py index 2c144917848..be355b3fe9c 100644 --- a/docs/zh_cn/conf.py +++ b/docs/zh_cn/conf.py @@ -57,6 +57,8 @@ myst_heading_anchors = 4 +myst_enable_extensions = ['colon_fence'] + autodoc_mock_imports = ['mmcv._ext', 'mmcv.utils.ext_loader', 'torchvision'] autosectionlabel_prefix_document = True diff --git a/docs/zh_cn/get_started/build.md b/docs/zh_cn/get_started/build.md index ec6ebb88794..a35be59bd3a 100644 --- a/docs/zh_cn/get_started/build.md +++ b/docs/zh_cn/get_started/build.md @@ -1,132 +1,199 @@ ## 从源码编译 MMCV -### 在 Linux 或者 macOS 上编译 MMCV +### 编译 mmcv-full -克隆算法库 +在编译 mmcv-full 之前,请确保 PyTorch 已经成功安装在环境中,可以参考 [PyTorch 官方安装文档](https://pytorch.org/get-started/locally/#start-locally)。可使用以下命令验证 ```bash -git clone https://github.com/open-mmlab/mmcv.git -cd mmcv +python -c 'import torch;print(torch.__version__)' ``` -建议安装 `ninja` 以加快编译速度 - -```bash -pip install -r requirements/optional.txt +```{note} +- 如需编译 ONNX Runtime 自定义算子,请参考[如何编译ONNX Runtime自定义算子](https://mmcv.readthedocs.io/zh_CN/latest/deployment/onnxruntime_op.html#id1) +- 如需编译 TensorRT 自定义,请参考[如何编译MMCV中的TensorRT插件](https://mmcv.readthedocs.io/zh_CN/latest/deployment/tensorrt_plugin.html#id3) ``` -你可以安装 lite 版本 +:::{note} + +- 如果克隆代码仓库的速度过慢,可以使用以下命令克隆(注意:gitee 的 mmcv 不一定和 github 的保持一致,因为每天只同步一次) ```bash -pip install -e . +git clone https://gitee.com/open-mmlab/mmcv.git ``` -也可以安装 full 版本 +- 如果打算使用 `opencv-python-headless` 而不是 `opencv-python`,例如在一个很小的容器环境或者没有图形用户界面的服务器中,你可以先安装 `opencv-python-headless`,这样在安装 mmcv 依赖的过程中会跳过 `opencv-python`。 + +- 如果编译过程安装依赖库的时间过长,可以[设置 pypi 源](https://mirrors.tuna.tsinghua.edu.cn/help/pypi/) ```bash -MMCV_WITH_OPS=1 pip install -e . +pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple ``` -如果是在 macOS 上编译,则需要在安装命令前添加一些环境变量 +::: -```bash -CC=clang CXX=clang++ CFLAGS='-stdlib=libc++' -``` +#### 在 Linux 上编译 mmcv-full -例如 +| TODO: 视频教程 -```bash -CC=clang CXX=clang++ CFLAGS='-stdlib=libc++' MMCV_WITH_OPS=1 pip install -e . -``` +1. 克隆代码仓库 -```{note} -如果你打算使用 `opencv-python-headless` 而不是 `opencv-python`,例如在一个很小的容器环境或者没有图形用户界面的服务器中,你可以先安装 `opencv-python-headless`,这样在安装 mmcv 依赖的过程中会跳过 `opencv-python` -``` + ```bash + git clone https://github.com/open-mmlab/mmcv.git + cd mmcv + ``` -### 在 Windows 上编译 MMCV +2. 安装 `ninja` 和 `psutil` 以加快编译速度 -在 Windows 上编译 MMCV 比 Linux 复杂,本节将一步步介绍如何在 Windows 上编译 MMCV。 + ```bash + pip install -r requirements/optional.txt + ``` -#### 依赖项 +3. 检查 nvcc 的版本(要求大于等于 9.2,如果没有 GPU,可以跳过) -请首先安装以下的依赖项: + ```bash + nvcc --version + ``` -- [Git](https://git-scm.com/download/win):安装期间,请选择 **add git to Path** -- [Visual Studio Community 2019](https://visualstudio.microsoft.com):用于编译 C++ 和 CUDA 代码 -- [Miniconda](https://docs.conda.io/en/latest/miniconda.html):包管理工具 -- [CUDA 10.2](https://developer.nvidia.com/cuda-10.2-download-archive):如果只需要 CPU 版本可以不安装 CUDA,安装CUDA时,可根据需要进行自定义安装。如果已经安装新版本的显卡驱动,建议取消驱动程序的安装 + 上述命令如果输出以下信息,表示 nvcc 的设置没有问题,否则需要设置 CUDA_HOME -```{note} -您需要知道如何在 Windows 上设置变量环境,尤其是 "PATH" 的设置,以下安装过程都会用到。 -``` + ``` + nvcc: NVIDIA (R) Cuda compiler driver + Copyright (c) 2005-2020 NVIDIA Corporation + Built on Mon_Nov_30_19:08:53_PST_2020 + Cuda compilation tools, release 11.2, V11.2.67 + Build cuda_11.2.r11.2/compiler.29373293_0 + ``` -#### 设置 Python 环境 + :::{note} + 如果想要支持 ROCm,可以参考 [AMD ROCm](https://rocmdocs.amd.com/en/latest/Installation_Guide/Installation-Guide.html) 安装 ROCm。 + ::: -1. 从 Windows 菜单启动 Anaconda 命令行 +4. 检查 gcc 的版本(要求大于等于**5.4**) -```{note} -如 Miniconda 安装程序建议,不要使用原始的 `cmd.exe` 或是 `powershell.exe`。命令行有两个版本,一个基于 PowerShell,一个基于传统的 `cmd.exe`。请注意以下说明都是使用的基于 PowerShell -``` + ```bash + gcc --version + ``` -2. 创建一个新的 Conda 环境 +5. 开始编译(预估耗时 10 分钟) - ```shell - conda create --name mmcv python=3.7 # 经测试,3.6, 3.7, 3.8 也能通过 - conda activate mmcv # 确保做任何操作前先激活环境 + ```bash + MMCV_WITH_OPS=1 pip install -e . -v ``` -3. 安装 PyTorch 时,可以根据需要安装支持 CUDA 或不支持 CUDA 的版本 +6. 验证安装 - ```shell - # CUDA version - conda install pytorch torchvision cudatoolkit=10.2 -c pytorch - # CPU version - conda install pytorch torchvision cpuonly -c pytorch + ```bash + python .dev_scripts/check_installation.py ``` -4. 准备 MMCV 源代码 + 如果上述命令没有报错,说明安装成功。如有报错,请查看[问题解决页面](https://mmcv.readthedocs.io/zh_CN/latest/faq.html)是否已经有解决方案。 + + 如果没有找到解决方案,欢迎提 [issue](https://github.com/open-mmlab/mmcv/issues)。 - ```shell +#### 在 macOS 上编译 mmcv-full + +| TODO: 视频教程 + +```{note} +如果你使用的 mac 是 M1 芯片,请安装 PyTorch 的 nightly 版本,否则会遇到 [issues#2218](https://github.com/open-mmlab/mmcv/issues/2218) 中的问题。 +``` + +1. 克隆代码仓库 + + ```bash git clone https://github.com/open-mmlab/mmcv.git cd mmcv ``` -5. 安装所需 Python 依赖包 +2. 安装 `ninja` 和 `psutil` 以加快编译速度 - ```shell - pip3 install -r requirements/runtime.txt + ```bash + pip install -r requirements/optional.txt ``` -6. 建议安装 `ninja` 以加快编译速度 +3. 开始编译 ```bash - pip install -r requirements/optional.txt + MMCV_WITH_OPS=1 pip install -e . ``` -#### 编译与安装 MMCV +4. 验证安装 + + ```bash + python .dev_scripts/check_installation.py + ``` -MMCV 有三种安装的模式: + 如果上述命令没有报错,说明安装成功。如有报错,请查看[问题解决页面](../faq.md)是否已经有解决方案。 -1. Lite 版本(不包含算子) + 如果没有找到解决方案,欢迎提 [issue](https://github.com/open-mmlab/mmcv/issues)。 - 这种方式下,没有算子被编译,这种模式的 mmcv 是原生的 python 包 +#### 在 Windows 上编译 mmcv-full -2. Full 版本(只包含 CPU 算子) +| TODO: 视频教程 - 编译 CPU 算子,但只有 x86 将会被编译,并且编译版本只能在 CPU only 情况下运行 +在 Windows 上编译 mmcv-full 比 Linux 复杂,本节将一步步介绍如何在 Windows 上编译 mmcv-full。 -3. Full 版本(既包含 CPU 算子,又包含 CUDA 算子) +##### 依赖项 - 同时编译 CPU 和 CUDA 算子,`ops` 模块的 x86 与 CUDA 的代码都可以被编译。同时编译的版本可以在 CUDA 上调用 GPU +请先安装以下的依赖项: + +- [Git](https://git-scm.com/download/win):安装期间,请选择 **add git to Path** +- [Visual Studio Community 2019](https://visualstudio.microsoft.com):用于编译 C++ 和 CUDA 代码 +- [Miniconda](https://docs.conda.io/en/latest/miniconda.html):包管理工具 +- [CUDA 10.2](https://developer.nvidia.com/cuda-10.2-download-archive):如果只需要 CPU 版本可以不安装 CUDA,安装 CUDA 时,可根据需要进行自定义安装。如果已经安装新版本的显卡驱动,建议取消驱动程序的安装 + +```{note} +如果不清楚如何安装以上依赖,请参考[Windows 环境从零安装 mmcv-full](https://zhuanlan.zhihu.com/p/434491590)。 +另外,你需要知道如何在 Windows 上设置变量环境,尤其是 "PATH" 的设置,以下安装过程都会用到。 +``` ##### 通用步骤 -1. 设置 MSVC 编译器 +1. 从 Windows 菜单启动 Anaconda 命令行 + + 如 Miniconda 安装程序建议,不要使用原始的 `cmd.exe` 或是 `powershell.exe`。命令行有两个版本,一个基于 PowerShell,一个基于传统的 `cmd.exe`。请注意以下说明都是使用的基于 PowerShell + +2. 创建一个新的 Conda 环境 + + ```powershell + (base) PS C:\Users\xxx> conda create --name mmcv python=3.7 + (base) PS C:\Users\xxx> conda activate mmcv # 确保做任何操作前先激活环境 + ``` + +3. 安装 PyTorch 时,可以根据需要安装支持 CUDA 或不支持 CUDA 的版本 + + ```powershell + # CUDA version + (mmcv) PS C:\Users\xxx> conda install pytorch torchvision cudatoolkit=10.2 -c pytorch + # CPU version + (mmcv) PS C:\Users\xxx> conda install install pytorch torchvision cpuonly -c pytorch + ``` + +4. 克隆代码仓库 + + ```powershell + (mmcv) PS C:\Users\xxx> git clone https://github.com/open-mmlab/mmcv.git + (mmcv) PS C:\Users\xxx> cd mmcv + ``` + +5. 安装 `ninja` 和 `psutil` 以加快编译速度 + + ```powershell + (mmcv) PS C:\Users\xxx\mmcv> pip install -r requirements/optional.txt + ``` + +6. 安装 mmcv 依赖 + + ```powershell + (mmcv) PS C:\Users\xxx\mmcv> pip install -r requirements/runtime.txt + ``` + +7. 设置 MSVC 编译器 设置环境变量。添加 `C:\Program Files (x86)\Microsoft Visual Studio\2019\Community\VC\Tools\MSVC\14.27.29110\bin\Hostx86\x64` 到 `PATH`,则 `cl.exe` 可以在命令行中运行,如下所示。 - ```none - (base) PS C:\Users\xxx> cl + ```powershell + (mmcv) PS C:\Users\xxx\mmcv> cl Microsoft (R) C/C++ Optimizing Compiler Version 19.27.29111 for x64 Copyright (C) Microsoft Corporation. All rights reserved. @@ -137,57 +204,45 @@ MMCV 有三种安装的模式: 因为 PyTorch 将解析 `cl.exe` 的输出以检查其版本,只有 utf-8 将会被识别,你可能需要将系统语言更改为英语。控制面板 -> 地区-> 管理-> 非 Unicode 来进行语言转换。 -##### 安装方式一:Lite version(不包含算子) +##### 编译与安装 mmcv-full -在完成上述的公共步骤后,从菜单打开 Anaconda 命令框,输入以下命令 +mmcv-full 有两个版本: -```shell -# 激活环境 -conda activate mmcv -# 切换到 mmcv 根目录 -cd mmcv -# 安装 -python setup.py develop -# 检查是否安装成功 -pip list -``` +- 只包含 CPU 算子的版本 -##### 安装方式二:Full version(只编译 CPU 算子) + 编译 CPU 算子,但只有 x86 将会被编译,并且编译版本只能在 CPU only 情况下运行 -1. 完成上述的公共步骤 +- 既包含 CPU 算子,又包含 CUDA 算子的版本 -2. 设置环境变量 + 同时编译 CPU 和 CUDA 算子,`ops` 模块的 x86 与 CUDA 的代码都可以被编译。同时编译的版本可以在 CUDA 上调用 GPU - ```shell - $env:MMCV_WITH_OPS = 1 - $env:MAX_JOBS = 8 # 根据你可用CPU以及内存量进行设置 - ``` +###### CPU 版本 -3. 编译安装 +1. 设置环境变量 - ```shell - conda activate mmcv # 激活环境 - cd mmcv # 改变路径 - python setup.py build_ext # 如果成功, cl 将被启动用于编译算子 - python setup.py develop # 安装 - pip list # 检查是否安装成功 + ```powershell + (mmcv) PS C:\Users\xxx\mmcv> $env:MMCV_WITH_OPS = 1 ``` -##### 安装方式三:Full version(既编译 CPU 算子又编译 CUDA 算子) +2. 编译安装 -1. 完成上述的公共步骤 + ```powershell + (mmcv) PS C:\Users\xxx\mmcv> python setup.py build_ext # 如果成功, cl 将被启动用于编译算子 + (mmcv) PS C:\Users\xxx\mmcv> python setup.py develop # 安装 + ``` -2. 设置环境变量 +###### GPU 版本 - ```shell - $env:MMCV_WITH_OPS = 1 - $env:MAX_JOBS = 8 # 根据你可用CPU以及内存量进行设置 +1. 设置环境变量 + + ```powershell + (mmcv) PS C:\Users\xxx\mmcv> $env:MMCV_WITH_OPS = 1 ``` -3. 检查 `CUDA_PATH` 或者 `CUDA_HOME` 环境变量已经存在在 `envs` 之中 +2. 检查 `CUDA_PATH` 或者 `CUDA_HOME` 环境变量已经存在在 `envs` 之中 - ```none - (base) PS C:\Users\WRH> ls env: + ```powershell + (mmcv) PS C:\Users\xxx\mmcv> ls env: Name Value ---- ----- @@ -198,38 +253,91 @@ pip list 如果没有,你可以按照下面的步骤设置 - ```shell - $env:CUDA_HOME = "C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2" + ```powershell + (mmcv) PS C:\Users\xxx\mmcv> $env:CUDA_HOME = "C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2" # 或者 - $env:CUDA_HOME = $env:CUDA_PATH_V10_2 # CUDA_PATH_V10_2 已经在环境变量中 + (mmcv) PS C:\Users\xxx\mmcv> $env:CUDA_HOME = $env:CUDA_PATH_V10_2 # CUDA_PATH_V10_2 已经在环境变量中 ``` -4. 设置 CUDA 的目标架构 +3. 设置 CUDA 的目标架构 - ```shell - $env:TORCH_CUDA_ARCH_LIST="6.1" # 支持 GTX 1080 - # 或者用所有支持的版本,但可能会变得很慢 - $env:TORCH_CUDA_ARCH_LIST="3.5 3.7 5.0 5.2 6.0 6.1 7.0 7.5" + ```powershell + # 这里需要改成你的显卡对应的目标架构 + (mmcv) PS C:\Users\xxx\mmcv> $env:TORCH_CUDA_ARCH_LIST="7.5" ``` -```{note} -我们可以在 [here](https://developer.nvidia.com/cuda-gpus) 查看 GPU 的计算能力 -``` + :::{note} + 可以点击 [cuda-gpus](https://developer.nvidia.com/cuda-gpus) 查看 GPU 的计算能力,也可以通过 CUDA 目录下的 deviceQuery.exe 工具查看 -5. 编译安装 + ```powershell + (mmcv) PS C:\Users\xxx\mmcv> &"C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\extras\demo_suite\deviceQuery.exe" + Device 0: "NVIDIA GeForce GTX 1660 SUPER" + CUDA Driver Version / Runtime Version 11.7 / 11.1 + CUDA Capability Major/Minor version number: 7.5 + ``` - ```shell - $env:MMCV_WITH_OPS = 1 - $env:MAX_JOBS = 8 # 根据你可用CPU以及内存量进行设置 - conda activate mmcv # 激活环境 - cd mmcv # 改变路径 - python setup.py build_ext # 如果成功, cl 将被启动用于编译算子 - python setup.py develop # 安装 - pip list # 检查是否安装成功 + 上面的 7.5 表示目标架构。注意:需把上面命令的 v10.2 换成你的 CUDA 版本。 + ::: + +4. 编译安装 + + ```powershell + (mmcv) PS C:\Users\xxx\mmcv> python setup.py build_ext # 如果成功, cl 将被启动用于编译算子 + (mmcv) PS C:\Users\xxx\mmcv> python setup.py develop # 安装 ``` -```{note} -如果你的 PyTorch 版本是 1.6.0,你可能会遇到一些这个 [issue](https://github.com/pytorch/pytorch/issues/42467) 提到的错误,则可以参考这个 [pull request](https://github.com/pytorch/pytorch/pull/43380/files) 修改 本地环境的 PyTorch 源代码 + ```{note} + 如果你的 PyTorch 版本是 1.6.0,你可能会遇到一些 [issue](https://github.com/pytorch/pytorch/issues/42467) 提到的错误,你可以参考这个 [pull request](https://github.com/pytorch/pytorch/pull/43380/files) 修改本地环境的 PyTorch 源代码 + ``` + +##### 验证安装 + +```powershell +(mmcv) PS C:\Users\xxx\mmcv> python .dev_scripts/check_installation.py ``` -如果编译安装 mmcv 的过程中遇到了问题,你也许可以在 [Frequently Asked Question](../faq.html) 找到解决方法 +如果上述命令没有报错,说明安装成功。如有报错,请查看[问题解决页面](../faq.md)是否已经有解决方案。 +如果没有找到解决方案,欢迎提 [issue](https://github.com/open-mmlab/mmcv/issues)。 + +### 编译 mmcv + +如果你需要使用和 PyTorch 相关的模块,请确保 PyTorch 已经成功安装在环境中,可以参考 [PyTorch 官方安装文档](https://pytorch.org/get-started/locally/#start-locally)。 + +1. 克隆代码仓库 + + ```bash + git clone https://github.com/open-mmlab/mmcv.git + cd mmcv + ``` + +2. 开始编译 + + ```bash + pip install -e . -v + ``` + +3. 验证安装 + + ```bash + python -c 'import mmcv;print(mmcv.__version__)' + ``` + +### 在 IPU 机器编译 mmcv + +首先你需要有可用的 IPU 云机器,可以查看[这里](https://www.graphcore.ai/ipus-in-the-cloud)。 + +#### 选项1: 使用 Docker + +1. 拉取镜像 + + ```bash + docker pull graphcore/pytorch + ``` + +2. 编译 mmcv + +#### 选项2: 使用 SDK + +1. 编译 mmcv + +2. 参考 [IPU PyTorch document](https://docs.graphcore.ai/projects/poptorch-user-guide/en/latest/installation.html) 安装 sdk。 diff --git a/docs/zh_cn/get_started/installation.md b/docs/zh_cn/get_started/installation.md index a6a20b05418..878193bdb1e 100644 --- a/docs/zh_cn/get_started/installation.md +++ b/docs/zh_cn/get_started/installation.md @@ -2,179 +2,369 @@ MMCV 有两个版本: -- **mmcv-full**: 完整版,包含所有的特性以及丰富的开箱即用的 CUDA 算子。注意完整版本可能需要更长时间来编译。 -- **mmcv**: 精简版,不包含 CUDA 算子但包含其余所有特性和功能,类似 MMCV 1.0 之前的版本。如果你不需要使用 CUDA 算子的话,精简版可以作为一个考虑选项。 +- **mmcv-full**: 完整版,包含所有的特性以及丰富的开箱即用的 CPU 和 CUDA 算子。注意,完整版本可能需要更长时间来编译。 +- **mmcv**: 精简版,不包含 CPU 和 CUDA 算子但包含其余所有特性和功能,类似 MMCV 1.0 之前的版本。如果你不需要使用算子的话,精简版可以作为一个考虑选项。 ```{warning} 请不要在同一个环境中安装两个版本,否则可能会遇到类似 `ModuleNotFound` 的错误。在安装一个版本之前,需要先卸载另一个。`如果CUDA可用,强烈推荐安装mmcv-full`。 ``` -a. 安装完整版 +### 安装 mmcv-full -在安装 mmcv-full 之前,请确保 PyTorch 已经成功安装在环境中,可以参考 PyTorch 官方[文档](https://pytorch.org/)。 +```{note} +- 下述安装步骤仅适用于 Linux 和 Windows 平台,如需在 macOS 平台安装 mmcv-full,请参考[源码安装 mmcv-full](https://mmcv.readthedocs.io/zh_CN/latest/get_started/build.html#macos-mmcv-full)。 +- 如需编译 ONNX Runtime 自定义算子,请参考[如何编译ONNX Runtime自定义算子](https://mmcv.readthedocs.io/zh_CN/latest/deployment/onnxruntime_op.html#id1) +- 如需编译 TensorRT 自定义,请参考[如何编译MMCV中的TensorRT插件](https://mmcv.readthedocs.io/zh_CN/latest/deployment/tensorrt_plugin.html#id3) +``` + +在安装 mmcv-full 之前,请确保 PyTorch 已经成功安装在环境中,可以参考 [PyTorch 官方安装文档](https://pytorch.org/get-started/locally/#start-locally)。可使用以下命令验证 + +```bash +python -c 'import torch;print(torch.__version__)' +``` -我们提供了 **Linux 和 Windows 平台** PyTorch 和 CUDA 版本组合的 mmcv-full 预编译包,可以大大简化用户安装编译过程。强烈推荐通过预编译包来安装。另外,安装完成后可以运行 [check_installation.py](https://github.com/open-mmlab/mmcv/.dev_scripts/check_installation.py) 脚本检查 mmcv-full 是否安装成功。 +如果输出版本信息,则表示 PyTorch 已安装。 -i. 安装最新版本 +#### 使用 mim 安装(推荐) -如下是安装最新版 `mmcv-full` 的命令 +[mim](https://github.com/open-mmlab/mim) 是 OpenMMLab 项目的包管理工具,使用它可以很方便地安装 mmcv-full。 -```shell -pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/{cu_version}/{torch_version}/index.html +```bash +pip install -U openmim +mim install mmcv-full ``` -请将链接中的 `{cu_version}` 和 `{torch_version}` 根据自身需求替换成实际的版本号,例如想安装和 `CUDA 11.1`、`PyTorch 1.9.0` 兼容的最新版 `mmcv-full`,使用如下替换过的命令 +如果发现上述的安装命令没有使用预编译包(以 `.whl` 结尾)而是使用源码包(以 `.tar.gz` 结尾)安装,则有可能是我们没有提供和当前环境的 PyTorch 版本、CUDA 版本相匹配的 mmcv-full 预编译包,此时,你可以[源码安装 mmcv-full](build.md)。 + +
+使用预编译包的安装日志 + +Looking in links: https://download.openmmlab.com/mmcv/dist/cu102/torch1.8.0/index.html
+Collecting mmcv-full
+Downloading https://download.openmmlab.com/mmcv/dist/cu102/torch1.8.0/mmcv_full-1.6.1-cp38-cp38-manylinux1_x86_64.whl + +
+ +
+使用源码包的安装日志 + +Looking in links: https://download.openmmlab.com/mmcv/dist/cu102/torch1.8.0/index.html
+Collecting mmcv-full==1.6.0
+Downloading mmcv-full-1.6.0.tar.gz -```shell -pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/cu111/torch1.9.0/index.html +
+ +如需安装指定版本的 mmcv-full,例如安装 1.6.0 版本的 mmcv-full,可使用以下命令 + +```bash +mim install mmcv-full==1.6.0 ``` -```{note} -PyTorch 在 1.x.0 和 1.x.1 之间通常是兼容的,故 mmcv-full 只提供 1.x.0 的编译包。如果你 -的 PyTorch 版本是 1.x.1,你可以放心地安装在 1.x.0 版本编译的 mmcv-full。例如,如果你的 -PyTorch 版本是 1.8.1、CUDA 版本是 11.1,你可以使用以下命令安装 mmcv-full。 +:::{note} +如果你打算使用 `opencv-python-headless` 而不是 `opencv-python`,例如在一个很小的容器环境或者没有图形用户界面的服务器中,你可以先安装 `opencv-python-headless`,这样在安装 mmcv 依赖的过程中会跳过 `opencv-python`。 -`pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/cu111/torch1.8.0/index.html` +另外,如果安装依赖库的时间过长,可以指定 pypi 源 + +```bash +mim install mmcv-full -i https://pypi.tuna.tsinghua.edu.cn/simple ``` -如果想知道更多 CUDA 和 PyTorch 版本的命令,可以参考下面的表格,将链接中的 `=={mmcv_version}` 删去即可。 +::: + +安装完成后可以运行 [check_installation.py](https://github.com/open-mmlab/mmcv/.dev_scripts/check_installation.py) 脚本检查 mmcv-full 是否安装成功。 -ii. 安装特定的版本 +#### 使用 pip 安装 -如下是安装特定版本 `mmcv-full` 的命令 +使用以下命令查看 CUDA 和 PyTorch 的版本 -```shell -pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/{cu_version}/{torch_version}/index.html +```bash +python -c 'import torch;print(torch.__version__);print(torch.version.cuda)' ``` -首先请参考版本发布信息找到想要安装的版本号,将 `{mmcv_version}` 替换成该版本号,例如 `1.3.9`。 -然后将链接中的 `{cu_version}` 和 `{torch_version}` 根据自身需求替换成实际的版本号,例如想安装和 `CUDA 11.1`、`PyTorch 1.9.0` 兼容的 `mmcv-full` 1.3.9 版本,使用如下替换过的命令 +根据系统的类型、CUDA 版本、PyTorch 版本以及 MMCV 版本选择相应的安装命令 + + + + +
+ + + + +
+

+
+
+
+
+如果在上面的下拉框中没有找到对应的版本,则可能是没有对应 PyTorch 或者 CUDA 或者 mmcv-full 版本的预编译包,此时,你可以[源码安装 mmcv-full](../faq.md)。
 
-```shell
-pip install mmcv-full==1.3.9 -f https://download.openmmlab.com/mmcv/dist/cu111/torch1.9.0/index.html
+:::{note}
+PyTorch 在 1.x.0 和 1.x.1 之间通常是兼容的,故 mmcv-full 只提供 1.x.0 的编译包。如果你
+的 PyTorch 版本是 1.x.1,你可以放心地安装在 1.x.0 版本编译的 mmcv-full。例如,如果你的
+PyTorch 版本是 1.8.1,你可以放心选择 1.8.x。
+:::
+
+:::{note}
+如果你打算使用 `opencv-python-headless` 而不是 `opencv-python`,例如在一个很小的容器环境或者没有图形用户界面的服务器中,你可以先安装 `opencv-python-headless`,这样在安装 mmcv 依赖的过程中会跳过 `opencv-python`。
+
+另外,如果安装依赖库的时间过长,可以指定 pypi 源
+
+```bash
+pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/cu111/torch1.9.0/index.html -i https://pypi.tuna.tsinghua.edu.cn/simple
 ```
 
-对于更多的 PyTorch 和 CUDA 版本组合,请参考下表:
-
-
-  
-    
-      
-      
-      
-      
-      
-      
-      
-      
-    
-    
-      
-      
-      
-      
-      
-      
-      
-      
-    
-    
-      
-      
-      
-      
-      
-      
-      
-      
-    
-    
-      
-      
-      
-      
-      
-      
-      
-      
-    
-    
-      
-      
-      
-      
-      
-      
-      
-      
-    
-    
-      
-      
-      
-      
-      
-      
-      
-      
-    
-    
-      
-      
-      
-      
-      
-      
-      
-      
-    
-    
-      
-      
-      
-      
-      
-      
-      
-      
-    
-    
-      
-      
-      
-      
-       
-      
-      
-      
-    
-  
-
CUDA torch 1.11torch 1.10torch 1.9torch 1.8torch 1.7torch 1.6torch 1.5
11.5
安装
pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cu115/torch1.11.0/index.html
11.3
安装
pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cu113/torch1.11.0/index.html
安装
pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cu113/torch1.10.0/index.html
11.1
安装
pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cu111/torch1.10.0/index.html
安装
pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cu111/torch1.9.0/index.html
安装
pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cu111/torch1.8.0/index.html
11.0
安装
pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cu110/torch1.7.0/index.html
10.2
安装
pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cu102/torch1.11.0/index.html
安装
pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cu102/torch1.10.0/index.html
安装
pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cu102/torch1.9.0/index.html
安装
pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cu102/torch1.8.0/index.html
安装
pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cu102/torch1.7.0/index.html
安装
pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cu102/torch1.6.0/index.html
安装
pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cu102/torch1.5.0/index.html
10.1
安装
 pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cu101/torch1.8.0/index.html
安装
 pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cu101/torch1.7.0/index.html
安装
 pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cu101/torch1.6.0/index.html
安装
 pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cu101/torch1.5.0/index.html
9.2
安装
 pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cu92/torch1.7.0/index.html
安装
 pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cu92/torch1.6.0/index.html
安装
 pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cu92/torch1.5.0/index.html
cpu
安装
 pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cpu/torch1.11.0/index.html
安装
 pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cpu/torch1.10.0/index.html
安装
 pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cpu/torch1.9.0/index.html
安装
 pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cpu/torch1.8.0/index.html
安装
 pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cpu/torch1.7.0/index.html
安装
 pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cpu/torch1.6.0/index.html
安装
 pip install mmcv-full=={mmcv_version} -f https://download.openmmlab.com/mmcv/dist/cpu/torch1.5.0/index.html
+::: -```{note} -以上提供的预编译包并不囊括所有的 mmcv-full 版本,我们可以点击对应链接查看支持的版本。例如,点击 [cu102-torch1.8.0](https://download.openmmlab.com/mmcv/dist/cu102/torch1.8.0/index.html),可以看到 `cu102-torch1.8.0` 只提供了 1.3.0 及以上的 mmcv-full 版本。另外,从 `mmcv v1.3.17` 开始,我们不再提供`PyTorch 1.3 & 1.4` 对应的 mmcv-full 预编译包。你可以在 [这](./previous_versions.md) 找到 `PyTorch 1.3 & 1.4` 对应的预编包。虽然我们不再提供 `PyTorch 1.3 & 1.4` 对应的预编译包,但是我们依然在 CI 中保证对它们的兼容持续到下一年。 +安装完成后可以运行 [check_installation.py](https://github.com/open-mmlab/mmcv/.dev_scripts/check_installation.py) 脚本检查 mmcv-full 是否安装成功。 + +#### 使用 docker 镜像 + +先将算法库克隆到本地再构建镜像 + +```bash +git clone https://github.com/open-mmlab/mmcv.git && cd mmcv +docker build -t mmcv -f docker/release/Dockerfile . ``` -```{note} -mmcv-full 没有提供 Windows 平台 `cu102-torch1.8.0` 和 `cu92-torch*` 的预编译包。 +也可以直接使用下面的命令构建镜像 + +```bash +docker build -t mmcv https://github.com/open-mmlab/mmcv.git#master:docker/release ``` -除了使用预编译包之外,另一种方式是在本地进行编译,直接运行下述命令 +[Dockerfile](release/Dockerfile) 默认安装最新的 mmcv-full,如果你想要指定版本,可以使用下面的命令 -```python -pip install mmcv-full +```bash +docker image build -t mmcv -f docker/release/Dockerfile --build-arg MMCV=1.5.0 . ``` -但注意本地编译可能会耗时 10 分钟以上。 +如果你想要使用其他版本的 PyTorch 和 CUDA,你可以在构建镜像时指定它们的版本。 -b. 安装精简版 +例如指定 PyTorch 的版本是 1.11,CUDA 的版本是 11.3 -```python -pip install mmcv +```bash +docker build -t mmcv -f docker/release/Dockerfile \ + --build-arg PYTORCH=1.11.0 \ + --build-arg CUDA=11.3 \ + --build-arg CUDNN=8 \ + --build-arg MMCV=1.5.0 . ``` -c. 安装完整版并且编译 onnxruntime 的自定义算子 +更多 PyTorch 和 CUDA 镜像可以点击 [dockerhub/pytorch](https://hub.docker.com/r/pytorch/pytorch/tags) 查看。 -- 详细的指南请查看 [这里](https://mmcv.readthedocs.io/zh_CN/latest/deployment/onnxruntime_custom_ops.html)。 +### 安装 mmcv -如果想从源码编译 MMCV,请参考[该文档](https://mmcv.readthedocs.io/zh_CN/latest/get_started/build.html)。 +如果你需要使用和 PyTorch 相关的模块,请确保 PyTorch 已经成功安装在环境中,可以参考 [PyTorch 官方安装文档](https://pytorch.org/get-started/locally/#start-locally)。 + +```python +pip install mmcv +``` diff --git a/docs/zh_cn/get_started/introduction.md b/docs/zh_cn/get_started/introduction.md index 99071325492..0fe46d05444 100644 --- a/docs/zh_cn/get_started/introduction.md +++ b/docs/zh_cn/get_started/introduction.md @@ -21,15 +21,15 @@ MMCV 是一个面向计算机视觉的基础库,它支持了很多开源项目 - [MMGeneration](https://github.com/open-mmlab/mmgeneration): OpenMMLab 图片视频生成模型工具箱 - [MMDeploy](https://github.com/open-mmlab/mmdeploy): OpenMMLab 模型部署框架 -MMCV 提供了如下众多功能: +MMCV 提供了以下功能: -- 通用的 IO 接口 -- 图像和视频处理 -- 图像和标注结果可视化 -- 常用小工具(进度条,计时器等) -- 基于 PyTorch 的通用训练框架 -- 多种 CNN 网络结构 -- 高质量实现的常见 CUDA 算子 +- [通用的 IO 接口](https://mmcv.readthedocs.io/zh_CN/latest/understand_mmcv/io.html) +- [图像和视频处理](https://mmcv.readthedocs.io/zh_CN/latest/understand_mmcv/data_process.html) +- [图像和标注结果可视化](https://mmcv.readthedocs.io/zh_CN/latest/understand_mmcv/visualization.html) +- [常用小工具(进度条,计时器等)](https://mmcv.readthedocs.io/zh_CN/latest/understand_mmcv/utils.html) +- [基于 PyTorch 的通用训练框架](https://mmcv.readthedocs.io/zh_CN/latest/understand_mmcv/runner.html) +- [多种 CNN 网络结构](https://mmcv.readthedocs.io/zh_CN/latest/understand_mmcv/cnn.html) +- [高质量实现的 CPU 和 CUDA 算子](https://mmcv.readthedocs.io/zh_CN/latest/understand_mmcv/ops.html) MMCV 支持以下的系统: @@ -37,7 +37,7 @@ MMCV 支持以下的系统: - Windows - macOS -如想了解更多特性和使用,请参考[文档](https://mmcv.readthedocs.io/zh_CN/latest)。 +欢迎查看[文档](http://mmcv.readthedocs.io/zh_CN/latest)了解更多特性和用法。 ```{note} MMCV 需要 Python 3.6 以上版本。 diff --git a/requirements/optional.txt b/requirements/optional.txt index 63730036fd3..bc74f1d295b 100644 --- a/requirements/optional.txt +++ b/requirements/optional.txt @@ -1 +1,2 @@ ninja +psutil