从创建conda环境再到安装mmcv及mmdet

1、cuda和cudnn

网上有很多,后期有需要再出吧。以前也踩了不少坑!

2、conda环境

创建conda环境
conda create -n pytorch python=3.8
移除conda环境
conda remove -n pytorch --all
激活conda环境
conda activate pytorch

3、安装pytorch

(以cuda10.2,pytorch1.9,python3.8为例)
由于官网下载太慢,我直接把包下载下来安装。下载地址如下:https://download.pytorch.org/whl/torch_stable.html
也可以到pytorch官网:https://pytorch.org/命令行安装
pip install torchvision-0.10.0+cu102-cp38-cp38-win_amd64.whl
pip install torchaudio-0.9.0-cp38-cp38-win_amd64.whl
pip install torch-1.9.0+cu102-cp38-cp38-win_amd64.whl

安装验证

命令行输入:
python
import torch
print(torch.cuda.is_available())
出现true说明安装成功
然后输入exit()退出

4、安装mmcv

一句话安装mmcv: https://github.com/open-mmlab/mmcv#installation
注:网址是可以点进去的。安装前可以看看有没有对应环境的包。
pip install mmcv-full==1.4.6 -f https://download.openmmlab.com/mmcv/dist/cu113/torch1.10.0/index.html

5、安装mmdet

自己编译mmdet的话,你编译的当前mmdet的repo就是你运行的mmdet。如果使用 pip install mmdet 来安装的话,会变成一个包安装在环境的sitepackges里面。在你之后修改当前代码是无效的。你需要修改环境中的那个mmdet。
自己git下来mmdet包,然后编译

git clone https://github.com/open-mmlab/mmdetection.git
cd mmdetection
pip install -r requirements/build.txt
pip install -v -e .  # or "python setup.py develop"

最后使用pip list。
你会看到mmcv装上了。mmdet后面会带一个路径,就是你当前编译的路径。

6、验证

1.在mmdetection文件夹中新建一个checkpoints文件夹
把这个网站下https://download.openmmlab.com/mmdetection/v2.0/faster_rcnn/faster_rcnn_r50_fpn_1x_coco/faster_rcnn_r50_fpn_1x_coco_20200130-047c8118.pth的文件放入其中。
2.把下面这段代码创建一个py文件到mmdetection文件夹中

from mmdet.apis import init_detector, inference_detector

config_file = 'configs/faster_rcnn/faster_rcnn_r50_fpn_1x_coco.py'
# download the checkpoint from model zoo and put it in `checkpoints/`
# url: https://download.openmmlab.com/mmdetection/v2.0/faster_rcnn/faster_rcnn_r50_fpn_1x_coco/faster_rcnn_r50_fpn_1x_coco_20200130-047c8118.pth
checkpoint_file = 'checkpoints/faster_rcnn_r50_fpn_1x_coco_20200130-047c8118.pth'
device = 'cuda:0'
# init a detector
model = init_detector(config_file, checkpoint_file, device=device)
# inference the demo image
inference_detector(model, 'demo/demo.jpg')

3.cd 到mmdetection文件夹,并执行 python test.py

你可能感兴趣的:(openmmlab,pytorch,python,深度学习)