RT-DETR的学习笔记

1. 复现训练流程

1.1 环境要求

paddlepaddle-gpu == 2.3.2

===========
= Summary =
===========

Driver:   Not Selected
Toolkit:  Installed in /usr/local/cuda-10.2/
Samples:  Installed in /root/, but missing recommended libraries

Please make sure that
 -   PATH includes /usr/local/cuda-10.2/bin
 -   LD_LIBRARY_PATH includes /usr/local/cuda-10.2/lib64, or, add /usr/local/cuda-10.2/lib64 to /etc/ld.so.conf and run ldconfig as root

To uninstall the CUDA Toolkit, run cuda-uninstaller in /usr/local/cuda-10.2/bin

Please see CUDA_Installation_Guide_Linux.pdf in /usr/local/cuda-10.2/doc/pdf for detailed information on setting up CUDA.
***WARNING: Incomplete installation! This installation did not install the CUDA Driver. A driver of version at least 440.00 is required for CUDA 10.2 functionality to work.
To install the driver using this installer, run the following command, replacing <CudaInstaller> with the name of this run file:
    sudo <CudaInstaller>.run --silent --driver

Logfile is /var/log/cuda-installer.log

1.1.1 RT-DETR原文使用的设备

RT-DETR的学习笔记_第1张图片

1.2 创建conda环境

conda create --name paddle python=3.10

1.3 安装PaddleDetection推荐的paddle版本: 2.3.2

python -m pip install paddlepaddle-gpu==2.3.2.post111 -f \
https://www.paddlepaddle.org.cn/whl/linux/mkl/avx/stable.html

你可能感兴趣的:(RT-DETR)