MMdetection环境搭建

目录

  • 本机环境
  • 1.安装ANACONDA
  • 2.用Conda创建虚拟环境
  • 3.激活虚拟环境
  • 4.安装pytorch
    • 安装pytorch
  • 5.安装MMCV
  • 6.安装MMdetection
  • 测试

本机环境

纯小白,本文用作个人笔记,记录下环境搭建的过程(搭了10+天!!!)。如果安装过程有什么问题,请自行解决,因为我也不太懂,啊哈哈哈哈

  • Windows10家庭中文版
  • 微星GF63 8RD
  • CPU:Intel® Core™ i7-8750H CPU @ 2.20GHz
  • 显卡:NVIDIA GeForce GTX 1050 Ti
  • CUDA 11.7.102 driver

1.安装ANACONDA

去ANACONDA官网下载软件,然后直接安装就行了,具体安装参考下面博客,自己能安装上CONDA就行,下面再介绍怎么用。
链接: 超详细Anaconda安装教程

2.用Conda创建虚拟环境

  • 打开“菜单”
  • 找到并用管理员权限打开“Anaconda Prompt(exe)”
  • 关键步骤,创建虚拟环境,在命令行输入:conda create -n open-mmlab python=3.8然后回车(python根据项目需要,自行修改,但是一定要具体到版本号。open-mmlab是环境名称,可以自己改)
  • Anaconda Prompt(exe)窗口不要关闭,后面安装都要在open-mmlab环境下安装。

3.激活虚拟环境

在Anaconda Prompt(exe)窗口输入:conda activate open-mmlab,回车执行,这样就进入到虚拟环境了。

4.安装pytorch

在安装pytorch之前,先检查自己电脑的CUDA版本号,如果版本号太低,先在NVIDIA官网安装最新CUDA驱动,关于CUDA的安装自己去博客上找教程,网上很多。我的CUDA版本号之前是CUDA11.2,安装pytorch版本号总是对不上,后来去官网安装的CUDA11.7.

安装pytorch

  • 在命令窗口输入:
conda install pytorch==1.11.0 torchvision==0.12.0 torchaudio==0.11.0 cudatoolkit=11.3 -c pytorch

MMdetection环境搭建_第1张图片

  • 注意:本作用的不是最新的版本号,考虑到的最新的不稳定,并且出问题不好查解决办法。具体的对应版本可以参考Pytorch官网。
    链接: Pytorch官网

5.安装MMCV

这个安装非常关键,我们直接安装MMCV-full版本,我安装的1.5.1版本,不是最新版也不是最旧版,追求稳
在窗口输入(先不要急着执行,先往下看)

pip install mmcv-full==1.5.1 -f https://download.openmmlab.com/mmcv/dist/cu113/torch1.11.0/index.html

回车执行
MMdetection环境搭建_第2张图片
这个MMCV-full的版本可以在github自己查,选择对应的cuda版本号和pytorch版本号,点install,就会出来一个pip,复制下来。cuda版本可以比自己电脑的cuda版本低。
MMdetection环境搭建_第3张图片
MMdetection环境搭建_第4张图片
{mmcv_version}这个需要修改,怎么改呢?我是直接换成了1.5.1,具体选择参考这个官方文档。
mmdetection链接: Openmmlab-MMCV-full版本号参考
MMdetection环境搭建_第5张图片

6.安装MMdetection

(1) 从官网下载MMdetection,保存到英文路径下,并解压。
MMdetection下载链接: github
MMdetection环境搭建_第6张图片
MMdetection环境搭建_第7张图片
(2) 复制文件路径
MMdetection环境搭建_第8张图片
(3)像我这样操作下,进入到mmdetection-masterd的根目录MMdetection环境搭建_第9张图片
(4)输入

pip install -r requirements/build.txt

MMdetection环境搭建_第10张图片

安装依赖包
(5)再接着输入

pip install -v -e.

MMdetection环境搭建_第11张图片

如果上面的报错,再试试下面这个,不放心两个都运行一遍

python setup.py develop

(6)安装cv2的支持包

pip install opencv-python

MMdetection环境搭建_第12张图片

测试

pycharm的环境配置就不说了,就是不说,就是玩儿,嘿嘿
我们直接在Anaconda Prompt(exe)窗口测试
对了
在这之前,先在mmdetection-master文件夹里面建立一个checkpoints文件夹,里面放什么?
从这个网址进去,把里面的pth文件下载下来,然后放到checkpoints文件、夹里面。
链接: 官方提供的模型
然后,我们在Anaconda Prompt(exe)窗口输入

python demo/image_demo.py demo/demo.jpg configs/faster_rcnn/faster_rcnn_r50_fpn_2x_coco.py checkpoints/faster_rcnn_r50_fpn_1x_coco_20200130-047c8118.pth

回车执行
MMdetection环境搭建_第13张图片
结果如下
MMdetection环境搭建_第14张图片
出现这个图片,就跟解除单身22年封印一样

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