许多人在安装开源计算机视觉框架mmcv时,常常因为版本问题而感到迷茫。以下,我结合自己的经验,为大家分享一套快速安装mmcv的实用方法。
问题困扰
在安装开源计算机视觉框架mmcv的过程中,版本不兼容问题往往是最棘手的。就拿我第一次尝试安装mmsegmentation来说,遭遇了连续的失败,每次都满怀信心却以安装失败告终。这种因版本不匹配引起的困扰,让众多像我这样的初学者感到十分烦恼。调查数据显示,超过七成的人在安装mmcv时都会遇到类似的版本问题。
conda create --name openmmlab python=3.8 -y
conda activate openmmlab
虚拟环境创建
nvidia-smi
为确保安装顺利进行,官方指南明确指出需先构建conda虚拟环境。安装前一晚,我坐在电脑前,严格按照指南操作。在终端输入指定指令,很快便成功创建了虚拟环境。这样的独立环境,相当于为后续安装提供了一个相对安全的“避风港”,有助于降低干扰风险。
安装Pytorch
安装Pytorch到虚拟环境时,版本匹配至关重要。我查看了电脑上的CUDA版本,发现是12.2。我以为只需安装相应版本的Torch即可。但安装时,mmcv版本不兼容的问题让我安装失败。经过多次尝试,我最终选择了CUDA11.7版本的Torch。尽管PyTorch已升级至2.X版,但鉴于与mmcv的兼容性问题,我最终还是决定使用1.13版。我在网上搜寻到了PyTorch早期版本的下载地址,页面下拉后,迅速找到了符合需求的PyTorch版本。
使用MIM安装MMCV
conda install pytorch==1.13.0 torchvision==0.14.0 torchaudio==0.13.0 pytorch-cuda=11.7 -c pytorch -c nvidia
步骤与官方指引一致,只是安装mmcv时需格外留意版本挑选。可依据已安装的CUDA和Torch版本来决定mmcv的版本。比如,我的CUDA是11.7,Torch是1.13,那么mmcv应选用2.0.0至2.1之间的版本。版本确定后,直接在终端输入正确的安装命令即可。挑选了正确的版本后,安装流程通常会很顺利。不会出现因版本不兼容导致的错误提示。
安装MMSegmentation
pip install -U openmim
mim install mmengine
装好mmcv后,依据官方手册,MMSegmentation的安装流程便正式启动。那天早上,我内心既期待又略感忐忑。仔细检查了每一步骤后,我小心翼翼地输入了安装命令。屏幕上代码快速闪过,直至出现安装成功的提示,我才终于松了口气。经过这一阶段,mmsegmentation的安装顺利完成,而且今后不会再遭遇令人头疼的版本兼容难题。
成功总结
按照之前讲的方法,特别要注意每个阶段版本匹配的问题。像mmsegmentation这样的工具,基于mmcv开发,安装起来很方便。只要你的CUDA版本跟我的一样,就可以依次运行那些指令进行安装。希望大家都能够避免遇到麻烦,顺利开启mmcv的学习之旅。在安装mmcv或mmsegmentation软件时,大家是否遇到了一些难以解决的难题?别忘了点赞并分享这篇文章给更多人看看!