文章目录
立于皓月之边,不弱星光之势
MMCV介绍
MMCV 开源框架从属于 OpenMMLab 系列,OpenMMLab 系列是专为计算机视觉不同方向建立统一而开放的代码库。目前已经开源了多个算法框架,例如物体检测的 MMDetection、语义分割的 MMSegmentation 和姿态估计的 MMPose 等等。
MMCV的安装
安装地址:https://download.openmmlab.com/mmcv/dist/cu113/torch1.10.0/index.html
修改链接中的cu113
这里需要修改cu113为自己的cuda版本
查看自己cuda版本的方法:打开命令行输入nvcc -V
我这里是11.7,则改为117。如果没有则可以找相邻版本。
修改链接中的torch1.10.0
这里需要修改torch1.10.0为自己的torch版本
-
查看自己torch版本的方法:打开命令行输入
conda env list
查看所有环境。 -
输入
activate pytorch1.13
进入虚拟环境 -
输入
pip list
查看torch版本 -
下载之后打开命令提示符先切入到虚拟环境再切换到下载目录
输入命令:pip install mmcv_full-1.7.1-cp39-cp39-win_amd64.whl
物体分类MMCLS
源码下载
https://github.com/open-mmlab/mmclassification