diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml index b8a9f4d24..fdc9fa98a 100644 --- a/.github/workflows/build.yml +++ b/.github/workflows/build.yml @@ -34,15 +34,15 @@ jobs: ] include: - torch: 1.5.1+cu101 - torch_version: torch1.5.1 + torch_version: torch1.5 torchvision: 0.6.1+cu101 mmcv: 1.5.0 - torch: 1.7.0+cu101 - torch_version: torch1.7.0 + torch_version: torch1.7 torchvision: 0.8.1+cu101 mmcv: 1.7.0 - torch: 1.8.0+cu101 - torch_version: torch1.8.0 + torch_version: torch1.8 torchvision: 0.9.0+cu101 mmcv: 1.8.0 @@ -101,7 +101,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 mmcv: 1.9.0 diff --git a/docs/en/get_started.md b/docs/en/get_started.md index 95b565c42..9e2f724d0 100644 --- a/docs/en/get_started.md +++ b/docs/en/get_started.md @@ -118,6 +118,7 @@ conda activate open-mmlab conda install pytorch==1.7.0 torchvision==0.8.0 cudatoolkit=10.1 -c pytorch -y # install the latest mmcv +# 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/cu101/torch1.7.0/index.html # install mmgeneration @@ -127,6 +128,13 @@ pip install -r requirements.txt pip install -v -e . ``` +To be noted that, 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. + +```shell +# We can ignore the micro version of PyTorch +pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/cu102/torch1.10/index.html +``` + ### Developing with multiple MMGeneration versions The train and test scripts already modify the `PYTHONPATH` to ensure the script uses the `MMGeneration` in the current directory.