1. RT-DETR
GitHub: PaddleDetection/tree/develop/configs/rtdetr
2. 复现训练流程
2.1 原文使用设备
2.2 环境要求
4*v100
cuda = 10.2
paddlepaddle-gpu >= 2.4.1
2.3 创建conda环境
conda create --name ppdet python=3.10
2.4 安装RT-DETR推荐的paddle版本
前往官网安装当前稳定的paddle版本,[paddle-stable];
2.5 PPDet的多卡训练依赖于NCCL (CUDA-10.2)
NCCL官网:nccl-legacy-downloads
2.6 Clone项目代码
git clone https://github.com/PaddlePaddle/PaddleDetection.git
若无法正常连接GitHub,使用以下命令clone代码
AIStudio
拉取develop分支
!git clone -b develop https://gitee.com/paddlepaddle/PaddleDetection.git