实战营笔记"/>
OpenMMLab AI实战营笔记
OpenMMLab AI实战营笔记-1
第一课目录
- OpenMMLab AI实战营笔记-1
- OpenMMLab简介
- 安装教程
- 先安装Pytorch
- 安装mmcv
- 安装mmcv-lite
- 安装完成
OpenMMLab简介
OpenMMLab是香港中文大学-商汤科技联合实验室MMLab开源的算法平台,包含众多SOTA计算机视觉算法。该平台在Github上不是一个单独项目,包括目标检测库MMDetection等库和数据集,涉及超过10个研究方向,开放超过100种算法和600种预训练模型,目前Github总星标超过1.7万。这些库大部分基于深度学习PyTorch框架,算法紧跟前沿,文档丰富,是CV方向系统性较强、社区活跃的开源平台。该平台包括MMEngine、MMCV、MIM、MMClassification、MMDetection、MMDetection3D、MMRotate、MMYOLO、MMSegmentation、MMOCR、MMPose、MMHuman3D、MMSelfSup、MMRazor、MMFewShot、MMAction2、MMTracking、MMFlow、MMEditing、MMGeneration和MMDeploy等项目。
安装教程
OpenMMLab提供了两个版本的MMCV:
-
mmcv:功能全面,内置各种CUDA操作,构建时间较长。
-
mmcv-lite:轻量版,没有CUDA操作但包含其他所有功能,类似于mmcv<1.0.0。当您不需要这些CUDA操作时,它非常有用。
注意:不要在同一环境中安装两个版本,否则可能会出现错误,例如ModuleNotFound。在可用CUDA的情况下,强烈建议安装完整版本。
先安装Pytorch
如果您需要使用与PyTorch相关的模块,请确保已按照PyTorch官方安装指南在您的环境中成功安装PyTorch。(先装好显卡驱动和CUDA,我默认大家都会装了哈,不赘述。)
pip3 install torch torchvision torchaudio --index-url
安装mmcv
在安装mmcv之前,请确保已按照PyTorch官方安装指南成功安装PyTorch。对于Apple Silicon用户,请使用PyTorch 1.13+。
安装mmcv的命令如下:
pip install -U openmim
mim install mmcv
如果您需要指定mmcv的版本,可以使用以下命令:
mim install mmcv==2.0.0
如果您发现上述安装命令没有使用以.whl结尾的预构建软件包,而是使用以.tar.gz结尾的源码包,则可能没有与PyTorch或CUDA或mmcv版本相对应的预构建软件包,此时您可以从源代码构建mmcv。
有关更多安装方法,请参阅安装文档。
安装mmcv-lite
pip install -U openmim
mim install mmcv-lite
安装完成
(OpenMMLab) PS C:\Users\liu\PycharmProjects\OpenMMLab> python
Python 3.10.11 | packaged by Anaconda, Inc. | (main, Apr 20 2023, 18:56:50) [MSC v.1916 64 bit (AMD64)] on win32
Type "help", "copyright", "credits" or "license" for more information.
>>> import mmcv
>>>
更多推荐
OpenMMLab AI实战营笔记
发布评论