本机环境
纯小白,本文用作个人笔记,记录下环境搭建的过程(搭了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
- 注意:本作用的不是最新的版本号,考虑到的最新的不稳定,并且出问题不好查解决办法。具体的对应版本可以参考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
回车执行
这个MMCV-full的版本可以在github自己查,选择对应的cuda版本号和pytorch版本号,点install,就会出来一个pip,复制下来。cuda版本可以比自己电脑的cuda版本低。
{mmcv_version}这个需要修改,怎么改呢?我是直接换成了1.5.1,具体选择参考这个官方文档。
mmdetection链接: Openmmlab-MMCV-full版本号参考
6.安装MMdetection
(1) 从官网下载MMdetection,保存到英文路径下,并解压。
MMdetection下载链接: github
(2) 复制文件路径
(3)像我这样操作下,进入到mmdetection-masterd的根目录
(4)输入
pip install -r requirements/build.txt
安装依赖包
(5)再接着输入
pip install -v -e.
如果上面的报错,再试试下面这个,不放心两个都运行一遍
python setup.py develop
(6)安装cv2的支持包
pip install opencv-python
测试
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
回车执行
结果如下
出现这个图片,就跟解除单身22年封印一样