Compiling CUDA operators？

IDEA-Research / DINO

[ICLR 2023] Official implementation of the paper "DINO: DETR with Improved DeNoising Anchor Boxes for End-to-End Object Detection"

Apache License 2.0

2.19k stars 243 forks source link

Compiling CUDA operators？ #207

Open narwyllllll opened 1 year ago

narwyllllll commented 1 year ago

Compiling CUDA operators的cd models/dino/ops python setup的时候报错torch.cuda.OutOfMemoryError: CUDA out of memory. 该怎么解决？ torch.cuda.OutOfMemoryError: CUDA out of memory. Tried to allocate 7.50 GiB (GPU 0; 14.75 GiB total capacity; 7.50 GiB already allocated; 7.13 GiB free; 7.50 GiB reserved in total by PyTorch)

1106X commented 1 year ago

调小batch/numwork

asadpackagex commented 10 months ago

When are you getting this error? While running the model or during training?

meirashaf commented 3 months ago

When are you getting this error? While running the model or during training?

I got the same error while compiling cuda operators. To be exact, it's while executing python test.py command

aditdesai commented 2 weeks ago

Did anyone find a solution to this?