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.78k stars 186 forks source link

pytorch #58

Open qmlzg opened 10 months ago

qmlzg commented 10 months ago

大佬 为什么我的torchvision=0.15.2能匹配的版本只能是CPU版本的torch，然后GPU版本的torch能匹配torchvision0.16，然后不能运行一直报错AttributeError: module 'torchvision' has no attribute 'disable_beta_transforms_warning'，大佬能帮帮孩子吗？我想知道你们实现的torch和torch vision的确切版本吗？谢谢大佬

lyuwenyu commented 10 months ago 
torch                    2.0.1
torchaudio               2.0.2
torchvision              0.15.2
zhizezhang commented 6 months ago

大佬请问一下版本都是一样的，为什么还是这个问题呢

1216560202 commented 5 months ago

w

我也是，请问您解决了吗

zhizezhang commented 5 months ago

w

我也是，请问您解决了吗

换的ultralytics-rtdetr库简单一些，环境好配