Anyone success in CUDA >= 11.0

fundamentalvision / Deformable-DETR

Deformable DETR: Deformable Transformers for End-to-End Object Detection.

Apache License 2.0

3.15k stars 513 forks source link

Closed imhgchoi closed 2 years ago

imhgchoi commented 2 years ago

Did anyone have success in running Deformable DETR on CUDA 11?

amirhesamyazdi commented 2 years ago

Did you fix your problem? Can you please elaborate how?

imhgchoi commented 2 years ago

I didn't actually fix it. It just somehow worked in a specific environment. Here's my torch versions and stuff.

Python 3.7.11 torch 1.10.0 torchvision 0.11.1 cudatoolkit 11.1 (This should align with your cuda version)

Honestly, I can't guarantee the above versions are what made this work. But anyways, best of luck!

shiyan-cui commented 2 years ago

Did you fix your problem now? Can you please elaborate how?

bimsarapathiraja commented 1 year ago

python 3.7 torch 1.7.0 cudatoolkit 11.0 worked