[Issue] error in ms_deformable_im2col_cuda: no kernel image is available for execution on the device

SHI-Labs / OneFormer

OneFormer: One Transformer to Rule Universal Image Segmentation, arxiv 2022 / CVPR 2023

https://praeclarumjj3.github.io/oneformer

MIT License

1.51k stars 135 forks source link

[Issue] error in ms_deformable_im2col_cuda: no kernel image is available for execution on the device #67

Closed momo1986 closed 1 year ago

momo1986 commented 1 year ago

My command ispython train_net.py --config-file configs/cityscapes/swin/oneformer_swin_large_bs16_90k.yaml --eval-only MODEL.IS_TRAIN False MODEL.WEIGHTS 250_16_swin_l_oneformer_cityscapes_90k.pth MODEL.TEST.TASK semantic

Is there an issue related to CUDA?

Thanks & Regards! Momo

praeclarumjj3 commented 1 year ago

Hi @momo1986, did you setup the CUDA kernel for the MSDeformAttn module? Please look at the last step in the installation instructions.

praeclarumjj3 commented 1 year ago

Feel free to re-open if you are still facing any issues.

momo1986 commented 1 year ago

Hi, my CUDA is 11.1, set CUDA_HOME to /usr/local/cuda-11.1.

Then I follows the installing structions.

Still has a problem: error in ms_deformable_im2col_cuda: no kernel image is available for execution on the device