search

lyuwenyu / RT-DETR

[CVPR 2024] Official RT-DETR (RTDETR paddle pytorch), Real-Time DEtection TRansformer, DETRs Beat YOLOs on Real-time Object Detection. 🔥 🔥 🔥
Apache License 2.0
1.64k stars 178 forks source link

关于GPU使用的问题 #308

Open TokaiTeio3 opened 1 month ago

TokaiTeio3 commented 1 month ago

作者大大您好，我想请问该在哪里设置使用的GPU信息，我使用的是4060 laptop，直接运行的话一直在使用GPU，也没有在代码中找到可以修改的地方

lyuwenyu commented 1 month ago

https://github.com/lyuwenyu/RT-DETR/blob/main/rtdetr_pytorch/src/core/config.py#L74-L75