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
ImportError: Cannot import 'detrex._C', therefore 'MultiScaleDeformableAttention' is not available #85
Comments
I install ninja by sudo apt install ninja-build and rebuild detectron2 by python setup.py build
python setup.py install and rebuild detrex
After that, the error is gone! |
Yes,
if export CUDA_HOME=/path/to/cuda11.3/ just an example |
@rentainhe Thank you for your replay. Python 3.10.4 (main, Mar 31 2022, 08:41:55) [GCC 7.5.0] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> from torch.utils.cpp_extension import CUDA_HOME
>>> print(CUDA_HOME)
/usr/local/cuda
>>> Maybe I haven't set the variable when I build it for the first time. export CUDA_HOME=/usr/local/cuda And then rebuild again. Thank you again! Now I can train on COCO correctly! |
You're welcome~, leave an issue any time if you need our help : ) |
I install the detrex==0.1.0 success. But when running the train_net, I get the error.
error info
Install log
The text was updated successfully, but these errors were encountered: