_C related error

[mhyeonsoo]

Hi, thanks for the great sources.

I am trying to run the demo on Ubuntu, but facing the issue as

ms_deform_attn.py:31: UserWarning: Failed to load custom C++ ops. Running on CPU mode Only!
NameError: name '_C' is not defined

Guess the same issue has been already raised, but just wanted to tell you that Ubuntu also shows a same case.

Thanks,

[rentainhe]

Sorry for not highlight some key of the installation, we need GPU runtime to compile some custom operator~ So maybe you can set CUDA_HOME and rebuild this repo

Here's an example:

export CUDA_HOME=/path/to/cuda-11.3/

[mhyeonsoo]

Thanks for your quick response @rentainhe ,
but it still show same error after I do

export CUDA_HOME=/path/to/cuda-11.8/

It seems like because of the device.
when I set the device as cpu, it works fine, but when adding device='gpu', it shows the error.

[SlongLiu]

Maybe you can try to reinstall the libraries.

[DanManN]

I had the same error. I recreated the virtual environment I was using and reinstalled the libraries after exporting the CUDA_HOME variable and it fixed it for me.

[mhyeonsoo]

I reinstalled the libraries in the new docker env. as you guys mentioned, and it works now!
Thanks for your help,