mmdetection安装记录(使用google colab安装)

本文记录了在Google Colaboratory上安装mmdetection的过程,包括安装Anaconda3、降低gcc版本到5.5、安装pytorch和cudatoolkit,以及在安装过程中遇到的问题和解决方案。通过验证和测试环境,确保mmdetection能够正常运行。

 

介绍

           Google Colaboratory是谷歌开放的一款研究工具,主要用于机器学习的开发和研究。它给广大的AI开发者提供了免费的GPU使用,可以在上面跑tensorflow等框架。

            mmdetection集成了多种目标检测算法,它的git地址,环境要求在官方文档中。我在Google Colaboratory上搭建环境的时候遇到了问题,记录如下。

安装过程

安装成功的环境

OS: Ubuntu18.04
CUDA:10.0
NCCL: 1.3.5
GCC(G++): 5.5

具体安装步骤

1.安装Anaconda3

!wget -q https://repo.anaconda.com/archive/Anaconda3-2019.10-Linux-x86_64.sh
!chmod +x Anaconda3-2019.10-Linux-x86_64.sh
!bash ./Anaconda3-2019.10-Linux-x86_64.sh -b -f -p /content/anaconda3
# set PATH environment variable
import os
os.environ['PATH'] = "/content/anaconda3/bin:" + os.environ['PATH']

2.安装gcc(g++5)

由于系统自带的gcc是为7.4.0,需要替换成5.5版本

!apt-get install gcc-5 g++-5
!cd /u
Google Colab 环境中下载安装 `mmdetection` 库可以通过以下步骤实现。`mmdetection` 是一个基于 PyTorch 的开源目标检测工具箱,由 OpenMMLab 开发维护。安装时需要确保依赖项正确安装,并且环境配置符合要求。 ### 安装步骤 1. **安装基础依赖** 首先需要安装 `mmcv`,这是 `mmdetection` 的核心依赖库。推荐使用预编译的 `mmcv` 版本以简化安装过程: ```bash pip install mmcv ``` 2. **安装 mmdetection** 可以通过 pip 直接安装 `mmdetection`: ```bash pip install mmdetection ``` 或者,如果需要使用最新功能或进行自定义修改,可以从 GitHub 源码安装: ```bash git clone https://github.com/open-mmlab/mmdetection.git cd mmdetection pip install -r requirements.txt pip install -v -e . ``` 3. **验证安装** 安装完成后,可以通过以下命令验证是否成功: ```python import mmdet print(mmdet.__version__) ``` 如果输出版本号,说明 `mmdetection` 已成功安装。 4. **配置环境** 如果使用 GPU 环境,确保 `mmdetection` 与当前安装PyTorch 和 CUDA 版本兼容。Colab 默认提供 CUDA 支持,但需要确认 PyTorch 是否正确识别了 GPU: ```python import torch print(torch.cuda.is_available()) ``` 如果输出 `True`,则表示 GPU 已就绪,`mmdetection` 可以充分利用 GPU 进行训练和推理。 ### 示例代码 以下是一个简单的示例,展示如何使用 `mmdetection` 进行目标检测: ```python from mmdet.apis import init_detector, inference_detector import mmcv # 加载预训练模型 config_file = 'configs/faster_rcnn/faster_rcnn_r50_fpn_1x_coco.py' checkpoint_file = 'checkpoints/faster_rcnn_r50_fpn_1x_coco_20200130-047c6664.pth' model = init_detector(config_file, checkpoint_file, device='cuda:0') # 进行推理 img = 'test.jpg' # 替换为你的图像路径 result = inference_detector(model, img) # 可视化结果 model.show_result(img, result, out_file='result.jpg') ``` 请确保在运行上述代码前上传图像文件 `test.jpg` 到 Colab 环境,并下载相应的模型权重文件 `faster_rcnn_r50_fpn_1x_coco_20200130-047c6664.pth`。 ### 注意事项 - **GPU 资源**:Colab 提供的 GPU 资源有限,建议合理管理内存使用,避免在训练时出现内存不足的问题。 - **依赖版本**:确保 `mmdetection` 与其依赖库(如 `mmcv`)的版本兼容,避免因版本不匹配导致的错误。
评论 18
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值