search

fundamentalvision / Deformable-DETR

Deformable DETR: Deformable Transformers for End-to-End Object Detection.
Apache License 2.0
3.14k stars 513 forks source link

Compiling CUDA operators？ #203

Open narwyllllll opened 1 year ago

narwyllllll commented 1 year ago

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)

Chenxu-666 commented 1 year ago

if you run test.py, you met with the problem. no problem. Just run model! because test.py is used to verify. This my experience.