mmcv-full 无痛安装方法

本文最后更新于:2022年8月8日 上午

本文记录 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

  • 下载链接格式

    1
    'https://download.openmmlab.com/mmcv/dist/' + CUDA_VERSION + '/' + TORCH_VERSION + '/index.html'
  • CUDA_VERSION 表示当前环境 cuda 版本,例如

    当前 Cuda 版本 字符串
    cuda 9.2 cu92
    11.1 cu111
    cpu cpu
  • TORCH_VERSION 表示当前环境 torch 版本,例如

    torch 小版本都归零

    当前 torch 版本 字符串
    torch 1.11.x torch1.11.0
    torch 1.10.x torch1.10.0
    torch1.7.x torch1.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 安装即可

参考资料


mmcv-full 无痛安装方法
https://www.zywvvd.com/notes/environment/cuda/mmcv-package-install/mmcv-package-install/
作者
Yiwei Zhang
发布于
2022年8月7日
许可协议