MMCV安装指南

MMCV是一个开源的计算机视觉库,广泛用于基于Pytorch的深度学习项目中。本教程主要介绍MMCV的安装方法。

首先需要确认已安装Pytorch,CUDA及驱动版本。然后pip安装mmcv或从源码编译安装。需要注意 PyTorch、CUDA和MMCV版本匹配问题。

安装验证可以import mmcv测试是否成功。常见问题包括CUDA版本不匹配、编译错误等。可检查环境变量设置,删除后重新编译安装。

MMCV 是一个面向计算机视觉的基础库,它提供了以下功能:

  • 图像和视频处理
  • 图像和标注结果可视化
  • 图像变换
  • 多种 CNN 网络结构
  • 高质量实现的常见 CUDA 算子

它这里支持很多的 OpenMMLab 项目,mmcv与pytorch和cuda有版本兼容的问题,你可以通过文档查看。安装 MMCV — mmcv 2.0.1 文档

目前网页打开是2.0.x版本,你可以通过下面方式去找到相应的版本:

MMCV安装指南_第1张图片 

在安装 mmcv 之前,请确保 PyTorch 已经成功安装在环境中,可以参考 PyTorch 官方安装文档。也可使用以下命令验证:

(torch1.8) C:\Users\asus>python
Python 3.8.17 (default, Jul  5 2023, 20:44:21) [MSC v.1916 64 bit (AMD64)] :: Anaconda, Inc. on win32
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> print(torch.__version__)
2.0.1+cu118
>>> print(torch.version.cuda)
11.8
>>>
  • python
  • import torch
  • print(torch.__version__)
  • print(torch.version.cuda)

这里我们就使用pip安装即可,选择合适的系统,对应的cuda、pytorch版本,以及MMCV的版本选择。即可得到安装命令 

MMCV安装指南_第2张图片

需要注意的是,PyTorch 在 1.x.0 和 1.x.1 之间通常是兼容的,故 mmcv 只提供 1.x.0 的编译包。如果你 的 PyTorch 版本是 1.x.1,你可以放心地安装在 1.x.0 版本编译的 mmcv。例如,如果你的 PyTorch 版本是 1.8.1,你可以放心选择 1.8.x。

cuda118、pytroch2.0,mmcv2.0.0安装指令:

pip install mmcv==2.0.0 -f https://download.openmmlab.com/mmcv/dist/cu118/torch2.0/index.html

如果你安装依赖库的时间过长,可以指定 pypi 源:

pip install mmcv==2.0.0 -f https://download.openmmlab.com/mmcv/dist/cu118/torch2.0/index.html -i https://pypi.tuna.tsinghua.edu.cn/simple

 

你可能感兴趣的:(#,优质教程,pytorch,人工智能,python)