Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Can a train Deformable-DETR on windows #96

Open
Aliweka2020 opened this issue Sep 16, 2021 · 2 comments
Open

Can a train Deformable-DETR on windows #96

Aliweka2020 opened this issue Sep 16, 2021 · 2 comments

Comments

@Aliweka2020
Copy link

Hi, I using a RTX2060 GPU to train Deformable-DETR on windows machine.
when I type the training command
python main.py --dataset_file coco --batch_size 1 --coco_path "D:\code\coco\images" --output_dir output
this error appears to me
ModuleNotFoundError: No module named 'MultiScaleDeformableAttention'
trying to combile Cuda, I got this error
error: [Errno 2] No such file or directory: 'C:\Users\Ali\Deformable-DETR-main\models\ops\build\temp.win-amd64-3.7\Release\build.ninja'

@wmnj
Copy link

wmnj commented Aug 25, 2022

hello,please have you solved it?

@Aliweka2020
Copy link
Author

yes I have solved it.
you can follow the instruction to build the CUDA, but I guess it require certian cuda version and c++ installed.
I have tried some mix and it works for me.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants