【AI】MMLab环境搭建
1.conda创建新环境
按照https://mmdetection.readthedocs.io/en/v2.28.0/get_started.html中介绍,可以先创建3.8版本conda环境
conda create --name openmmlab python=3.8 -y
conda activate openmmlab
2.安装torch环境
conda install pytorch torchvision -c pytorch
如果是cpu环境可以使用下面的命令
conda install pytorch torchvision cpuonly -c pytorch
3.安装MMCV
官方推荐使用mim来进行安装,然而在实际安装过程中遇到了缺少cl编译的问题,于是采用下载whl来进行安装
- mim安装方式(供参考)
pip install -U openmim
mim install mmcv-full
- whl安装方式
先从网站上下载跟自己环境版本相适应的whl安装文件
https://download.openmmlab.com/mmcv/dist/cu121/torch2.1.0/index.html
这里cu121代表cuda版本是12.1版本,torch2.1.0代表是torch2.1版本,可能因为版本更新问题导致找不到你需要的版本,可以使用相近的版本来进行替换
./mmcv-2.1.0-cp310-cp310-manylinux1_x86_64.whl
./mmcv-2.1.0-cp310-cp310-win_amd64.whl
./mmcv-2.1.0-cp311-cp311-manylinux1_x86_64.whl
./mmcv-2.1.0-cp311-cp311-win_amd64.whl
./mmcv-2.1.0-cp38-cp38-manylinux1_x86_64.whl
./mmcv-2.1.0-cp38-cp38-win_amd64.whl
./mmcv-2.1.0-cp39-cp39-manylinux1_x86_64.whl
./mmcv-2.1.0-cp39-cp39-win_amd64.whl
网页会显示上面的内容,选择自己cp代表python版本,win代表windows环境 linux代表Linux环境,按照自己的环境标准选择即可。
现在完成之后,在conda promot中cd到下载路径,确认当前的虚拟环境就是自己想要安装的环境,然后执行命令
pip install mmcv-2.1.0-cp38-cp38-win_amd64.whl
等待安装完成