Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Update the installation of MMCV #409

Merged
merged 2 commits into from
Jan 28, 2022
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
10 changes: 5 additions & 5 deletions .github/workflows/build.yml
Original file line number Diff line number Diff line change
Expand Up @@ -32,10 +32,10 @@ jobs:
torch: [1.5.1, 1.7.1]
include:
- torch: 1.5.1
torch_version: torch1.5.0
torch_version: torch1.5
torchvision: 0.6.1
- torch: 1.7.1
torch_version: torch1.7.0
torch_version: torch1.7
torchvision: 0.8.2
steps:
- uses: actions/checkout@v2
Expand Down Expand Up @@ -73,10 +73,10 @@ jobs:
torch: [1.5.1+cu101, 1.7.1+cu101]
include:
- torch: 1.5.1+cu101
torch_version: torch1.5.0
torch_version: torch1.5
torchvision: 0.6.1+cu101
- torch: 1.7.1+cu101
torch_version: torch1.7.0
torch_version: torch1.7
torchvision: 0.8.2+cu101

steps:
Expand Down Expand Up @@ -142,7 +142,7 @@ jobs:
torch: [1.9.0+cu102]
include:
- torch: 1.9.0+cu102
torch_version: torch1.9.0
torch_version: torch1.9
torchvision: 0.10.0+cu102

steps:
Expand Down
10 changes: 9 additions & 1 deletion docs/en/install.md
Original file line number Diff line number Diff line change
Expand Up @@ -58,7 +58,15 @@ The compatible MMTracking, MMCV, and MMDetection versions are as below. Please i
3. Install mmcv-full, we recommend you to install the pre-build package as below.

```shell
pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/cu101/torch1.6.0/index.html
# pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/{cu_version}/{torch_version}/index.html
pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/cu102/torch1.10.0/index.html
```

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.

```
# We can ignore the micro version of PyTorch
pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/cu102/torch1.10/index.html
```

See [here](https://mmcv.readthedocs.io/en/latest/get_started/installation.html) for different versions of MMCV compatible to different PyTorch and CUDA versions.
Expand Down
10 changes: 9 additions & 1 deletion docs/zh_cn/install.md
Original file line number Diff line number Diff line change
Expand Up @@ -54,7 +54,15 @@
3. 安装 mmcv-full,我们推荐您安装以下预构建包:

```shell
pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/cu101/torch1.6.0/index.html
# pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/{cu_version}/{torch_version}/index.html
pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/cu102/torch1.10.0/index.html
```

PyTorch 在 1.x.0 和 1.x.1 之间通常是兼容的,故 mmcv-full 只提供 1.x.0 的编译包。如果你的 PyTorch 版本是 1.x.1,你可以放心地安装在 1.x.0 版本编译的 mmcv-full。

```
# 我们可以忽略 PyTorch 的小版本号
pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/cu102/torch1.10/index.html
```

请参阅 [MMCV](https://github.com/open-mmlab/mmcv#install-with-pip) 了解不同版本的 MMCV 与不同版本的 PyTorch 和 CUDA 的兼容情况。同时,您可以使用以下命令从源码编译 MMCV:
Expand Down