- ubuntun20.04
- anaconda(python3.8)
- pytorch1.5(cuda10.1)
- mmvc(pytorch1.5+cuda10.1)
- gcc7
- nvidia驱动 cuda10.1
1.ubuntun安装
参考链接
不能用虚拟机,因为不能安装显卡
2.python安装-anaconda安装 bilibili
- 官网下载linux版 下载Anaconda3-2021.05-Linux-x86_64.sh
- 终端执行
bash Anaconda3-2021.05-Linux-x86_64.sh
一路enter+yes - 配置默认python:
- 安装vim
sudo apt install vim
- 编辑这个文件
vim ~/.bashrc
- 在这个文件最下加上这一句话
export PATH="/home/oem/anaconda3/bin:$PATH"
- 终端执行
source ~/.bashrc
- 终端输入检测
python3
linux命令 | 含义 |
---|---|
i | 进入输入状态 |
esc | 进入命令状态 |
:wq | 保存文件并退出vi/vim |
3.pytorch1.5(cuda10.1)(如果下载慢导致失败:配置清华源/回归最初源) bilibili
-从官网找到命令终端执行 -conda install pytorch==1.5.0 torchvision==0.6.0 cudatoolkit=10.1 -c pytorch
4.安装mmvc(注意正确版本pytorch1.5+cuda10.1) bilibili
终端执行pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/cu101/torch1.5.0/index.html
5.安装项目需要的依赖(在mmdetection-master的requirements.txt里的)
终端~/mmdetection/mmdetection-master执行 pip install -r requirements.txt
或者~/mmdetection/mmdetection-master执行pip install -r requirements/build.txt
6.安装gcc5+(否则下面那个coocapi出错)
终端执行sudo apt-get install build-essential
7.cocoapi(看见tools就是配置成功) bilibili
- github里搜索cocoapi 下载
- 只需要pythonapi里面打开自己 建的续集环境的终端输入
python setu.py build_ext install
- 运行
pip list
看到pycocotools就是安装成功
8.mmdetection安装
- 在mmdetection-matser根目录运行
python setup.py develop
进行安装 pip list
看到mmdet 配置成功
9.运行demo
- 建立checkpoints 进入config-mask-下载模型
python demo/image_demo.py demo/demo.jpg configs/mask_rcnn/mask_rcnn_r50_fpn_2x_coco.py checkpoints/mask_rcnn_r50_fpn_2x_coco_bbox_mAP-0.392__segm_mAP-0.354_20200505_003907-3e542a40.pth
10.手动nvidia驱动
参考文章
- 禁用nouveau
- 删除已经有的驱动
- 按照显卡下载适合的驱动
- 出现了问题:禁用UEFI参考文章
- sudo reboot重启
- modprobe 加载模块时 出现权限不允许提醒;Operation not permitted参考文章
11.安装cuda
CUDA是NVIDIA推出的用于自家GPU的并行计算框架
CUDA只能在NVIDIA的GPU上运行
参考文章