[Python知识库]mmdetection安装笔记

查看版本

• 查看python版本
  python3.7
• 查看cuda版本
  
  cuda版本11.0（一般安装命令中就是cu110）

创建虚拟环境

# 联网
conda create -n env_name python = 3.7 anaconda # 带anaconda安装会自动安装很多常用的包，否则需要自己下载
#不联网
conda create -n env_name python = 3.7 --offline
# 复制已有的虚拟环境
conda create --name my_env_name --clone env_name

安装pytorch

# 卸载原有的pytorch，如果原来的pytorch的cuda版本和你要装的不符合，可以卸载
pip uninstall torch torchvision -y

# 结合上面的版本，下面命令二选一
pip install torch==1.7.0+cu110 torchvision==0.8.1+cu110 -f https://download.pytorch.org/whl/torch_stable.html
# -c pytorch是使用国外的源安装，建议去掉，改用清华源镜像
conda install pytorch==1.7.0 torchvision==0.8.1 cudatoolkit=11.0 -c pytorch

安装mmdetection

# 安装mmdetection
rm -rf mmdetection # 先删除已有的mmdetection，没有就不用删
git clone https://github.com/open-mmlab/mmdetection.git # 克隆mmdetection仓库到本地
cd mmdetection # 进入文件夹

pip install -r requirements/build.txt
pip install -v -e . 

# 安装mmcv-full
pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/cu110/torch1.7.0/index.html