Pytorch-Correlation-extension
Pytorch-Correlation-extension copied to clipboard
Unknown CUDA Arch (8.6) or GPU not supported
Hi, It looks like this module cannot be compiled in GTX 3090 GPU. I was wondering if there is a way to address this issue? Thank you very much!
Hi, not an owner of GTX3090 so I will struggle to help you properly, but you can have a look at this issue : #61 From what I understood, the solution is simply to update to pytorch nightly.
Good luck !
i meet the same problem ,have you slove it?? thanks!!!