mmpretrainmmdetection环境配置
mmpretrain&mmdetection环境配置
适用于cuda11.3+torch12.1的mmpretrain&mmdetection环境配置:
第一步:根据官网说明,找到对应cuda版本的torch,安装好torch:
pip install torch==1.12.1+cu113 torchvision==0.13.1+cu113 torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cu113
第二步:根据官网说明,安装对应cuda版本和torch版本的mmcv:
pip install mmcv==2.1.0 -f https://download.openmmlab.com/mmcv/dist/cu113/torch1.12/index.html
第三步:安装openmim,并使用mim安装mmpretrain:
pip install -U openmim
mim install "mmpretrain>=1.0.0rc8" #基础版本
mim install "mmpretrain[multimodal]>=1.0.0rc8" # 多模态版本
mmdetection安装类似,需要注意的是就是先根据cuda版本安装torch,然后根据两者版本安装openmmlab系列软件即可~
有一个遇到的bug很奇怪,就是目前2.1.2版本的numpy对这些版本比较老的torch和openmmlab系列软件不适用,需要降级到1.26.4版本~