安装流程

注:以一下版本为例

1. python 3.7
2. pytorch 1.6.0 torchvision 0.7.0
3. mmcv-full 1.2.0
4. cudatoolkit=10.1

1.创建conda虚拟环境并激活

conda create -n yyds(环境名称) python=3.7 -y
conda activate yyds

2.安装Pytorch和torchvison

其他对应版本点击here.

conda install pytorch==1.6.0 torchvision==0.7.0 cudatoolkit=10.1 -c pytorch

3.安装mmcv-full

其他对应版本点击here.

 pip install mmcv-full==1.2.0 -f https://download.openmmlab.com/mmcv/dist/cu101/torch1.6.0/index.html

4.安装相关环境

pip install -r requirements/build.txt
pip install -v -e .  # or "python setup.py develop"

大体流程如上,如有错误疏漏,请评论区留言