Windows10系统配置mmdetection2.6避坑指南

Windows10系统配置mmdetection2.6避坑指南

因为最近在参加2022年航行器大赛（C3-1海上智能感知组），选择Varifocalnet做训练，模型需要在mmdetection上配置，这篇博客主要记录在配置mmdetection2.6时遇到的一些问题以及解决方法。

版本说明：Python3.7.13+PyTorch1.6.0+CUDA10.1+CuDNN7.6.4.38+TorchVision0.7.0+MMCV1.1.5+Visual Studio2017+opencv-python4.2.0.34

1 安装CUDA

• CUDA 下载地址：CUDA Toolkit Archive | NVIDIA Developer,选择对应版本的CUDA（CUDA Toolkit 10.1）。

• 找到下载的驱动文件，双击打开后，保持默认就好（安装选项选择“精简”，不安装 Visual Studio）。

• 在 cmd 终端输入 nvcc -V 指令，测试 CUDA 是否安装成功。

2 安装CuDNN

• CuDNN 下载地址：cuDNN Archive | NVIDIA Developer，选择对应版本的 CuDNN, 选择 cuDNN Library for

  Windows (x86)，然后下载cudnn-10.1-windows10-x64-v7.6.4.38.zip。

• 下载完成后解压缩。里面有 bin、include、lib 三个目录。

• 将三个文件夹内容复制到安装 CUDA 安装目录，覆盖对应文件夹内容：CUDA 默认文件夹在：C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1。

3 下载mmdetection-2.6.0源码

• 链接

4 配置过程

（1）打开cmd，创建虚拟环境

conda create -n md26 python=3.7 -y

（2）激活虚拟环境

conda activate md26

（3）安装pytorch

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

也可以根据自己的版本进入官网自行安装。

（4）安装mmcv-full

• 大家可以先按照命令安装看看能不能成功，不成功的话继续往下看。

• 打开这个网页，找到mmcv-full1.1.5，对应的版本，这里注意cuda的版本跟自己的电脑对应。

# 我的版本是
mmcv_full-1.1.5+torch1.6.0+cu101-cp37-cp37m-win_amd64.whl

• cd 到该文件存放的路径中（我直接放在了D盘根目录下），输入下面的pip命令

pip install D:\mmcv_full-1.1.5+torch1.6.0+cu101-cp37-cp37m-win_amd64.whl

• 查看Packeage的版本

pip list

（5）安装requprements.txt

这个地方和原先下载的源码有极大关系。检查自己下载的源码是否正确，在这个地方踩了坑。

• 打开你的编译器，我用的vscode。pycharm也可以。在mmd26虚拟环境下打开终端，输入：

pip install -r requirements.txt

（6）最后在命令窗口下输入python setup.py develop安装完成。

python setup.py develop

（7）别忘记安装opencv-python

pip install opencv-python==4.2.0.34

5 运行demo

# 识别图片
python demo/image_demo.py demo/demo.jpg configs/faster_rcnn/faster_rcnn_r50_fpn_1x_coco.py checkpoints/faster_rcnn_r50_fpn_1x_coco_20200130-047c8118.pth

 

OK，到这里mmd2.6环境就配置结束了，有问题欢迎大家指正！过程还是比较简单，但是注意踩坑，掉进去赶紧出来！！！