说一下依托关系:MMCV是面向计算机视觉的一个基础库,它支持OpenMMLab的各个模块包括MMClassification图像分类,MMDetectionm目标检测,MMOCR文字检测识别等等,本文主要详细介绍一下mmcv和mmcls的安装。
目录
一.MMCV 安装
MMCV 有两个版本:
-
mmcv-full: 完整版,包含所有的特性以及丰富的开箱即用的 CPU 和 CUDA 算子。注意,完整版本可能需要更长时间来编译。
-
mmcv: 精简版,不包含 CPU 和 CUDA 算子但包含其余所有特性和功能,类似 MMCV 1.0 之前的版本。如果你不需要使用算子的话,精简版可以作为一个考虑选项。
1.查看CUDA 版本、PyTorch 版本
# 进入python命令行
python
# 这里会输出你的python版本
import torch
print(torch.__version__)
print(torch.version.cuda)