mmcv-full 無痛安裝方法

本文記錄 mmcv-full 1.3.0 以及之後的版本無痛安裝方法。

背景

• mmcv-full 安裝在之前需要自己編譯可執行檔案，煞費苦心配置環境還可能經常出問題
• 最近發現官方推出了無痛版安裝包，可以直接安裝編譯好的檔案包

下載下傳位址

• 不同環境下的安裝包在不同的目錄
• 主要區分 Cuda 版本和 torch 版本
• 官方連結

CUDA	torch 1.11	torch 1.10	torch 1.9	torch 1.8	torch 1.7	torch 1.6	torch 1.5
11.5	√
11.3	√	√
11.1	√	√	√
11.0	√
10.2	√	√	√	√	√	√	√
10.1	√	√	√	√
9.2	√	√	√
cpu	√	√	√	√	√	√	√

• 表格（更新至 2022-8-7）

  √

  部分表明存在該版本

  mmcv-full

  包
• 下載下傳連結格式

• CUDA_VERSION

  表示目前環境 cuda 版本，例如

  目前 Cuda 版本字元串cuda 9.2cu9211.1cu111cpucpu

• TORCH_VERSION

  表示目前環境 torch 版本，例如

  torch 小版本都歸零

  目前 torch 版本字元串torch 1.11.xtorch1.11.0torch 1.10.xtorch1.10.0torch1.7.xtorch1.7.0

• 就可以拼湊出下載下傳 mmcv-full 包的連結位址
• 例如 在 CUDA 11.1， torch1.8.2 版本下，下載下傳位址為 https://download.openmmlab.com/mmcv/dist/cu111/torch1.8.0/index.html

• 選擇自己需要的

  mmcv-full 版本

  、

  python 版本

  、

  作業系統

  下的安裝包
• 使用

  pip install file-path

  安裝即可

參考資料

• https://github.com/open-mmlab/mmcv