【软硬件环境安装】openMMLab的软件环境搭建

 

使用conda安装虚拟环境

首先，需要安装NVIDIA驱动和Conda工具；

conda create -n open_mmlab python=3.8 -y
conda activate open_mmlab
conda install pytorch=1.6.0 torchvision cudatoolkit=10.1 -c pytorch
pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/cu101/torch1.6.0/index.html
git clone https://github.com/open-mmlab/mmsegmentation.git
cd mmsegmentation/
pip install -e .

注意，根据主机的硬件显卡选择合适的驱动，并根据conda和驱动的版本选择torch和cuda的版本，一定要注意版本的匹配；

删除虚拟环境

conda remove -n open_mmlab --all

如果mmcv的版本不兼容，需要先删除原来版本的mmcv，再安装合适的版本；

pip uninstall mmcv

 

 

参考

1. openmmlab_get_started;

完