Faster-RCNN模型跑通总结（使用pytorch1.10+cuda10.2版本）

Faster-RCNN模型搭建跑通总结

• 0、前言
• 1、准备操作系统
• 2、安装驱动及cuda
• • 2.1、安装驱动
  • 2.2、安装cuda
• 3、安装anaconda和pytorch
• • 3.1 安装anaconda
  • • 3.1.1为什么推荐安装anaconda而不是pip安装？
    • 3.1.2 安装anaconda
    • 3.1.3 配置国内镜像源
  • 3.2 安装pytorch
  • • 3.2.1 确认要安装的pytorch版本
    • 3.2.2 安装pytorch
    • • 3.2.2.1 在conda中创建虚拟环境
      • 3.2.2.2 激活该虚拟环境
      • 3.2.2.3 在该环境中安装pytorch
      • 3.2.2.4 确认安装的pytorch版本
• 4、训练faster-rcnn模型
• • 4.1 下载模型到本地
  • 4.2 安装所需的依赖包
  • 4.3 编译cuda 依赖项
  • • 4.3.1 编译coco api
    • 4.3.2 编译cuda 依赖项
    • 4.3.3 验证是否编译成功
  • 4.4 下载数据集
  • 4.5 下载预训练模型
  • 4.6 训练模型
• 5、总结

0、前言

最近在学习Faster-RCNN模型，想在自己个人的笔记本上复现代码，本以为没那么复杂，但过程中却没那么简单，反倒是十分波折