这里,默认已安装好cuda以及anaconda相关环境了。先附上mmdetection的官方github链接以及MMDetection 2官方文档
一、创建anaconda虚拟环境
conda create -n mmdetection python=3.7 #创建环境
conda activate mmdetection #激活环境
二、安装pytorch
打开pytorch官网,查看对应的安装命令。举安装cuda10.2对应的版本为例:
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch