![3b777bc1d158b2e9b1c0959cfed401a7.png](https://img-blog.csdnimg.cn/img_convert/3b777bc1d158b2e9b1c0959cfed401a7.png)
原文GitHub地址[1]
环境搭建:
- Linux (Ubuntu 16.04/18.04 and CentOS 7.2)(Windows可以自己研究搭建,官方暂不支持)
- Python 3.5+ (Python 2不支持)
- PyTorch 1.1 or higher
- CUDA 9.0 or higher(9.0/9.2/10.0)
- NCCL 2(2.1.15/2.2.13/2.3.7/2.4.2)
- GCC(G++) 4.9 or higher(4.9/5.3/5.4/7.3)
- mmcv[2]
安装步骤:
- 创建虚拟环境:
conda create -n open-mmlab python=3.7
#如果需要移除虚拟环境
#conda remove -n open-mmlab --all
source activate
conda activate open-mmlab
2. 安装torch环境
conda install pytorch torchvision -c