Closed jeya-maria-jose closed 1 year ago
I had the same problem.
Which cuda and pytorch versions do you use? I currently use CUDA11.3+Pytorch1.12
.
which gcc version do you use? @JingyunLiang
Did you solve this problem? I also got the same error. @jeya-maria-jose
pip3 install torch==1.9.1+cu111 torchvision==0.10.1+cu111 torchaudio==0.9.1 -f https://download.pytorch.org/whl/torch_stable.html
These versions solved the dependency issue for me.
There seems to be a dependency issue of cuda kernel while using deform_attn while training if the current dependencies were installed.