Package has issues with Pytorch 1.10

[codtiger]

Hi,

As I was configuring different combinations of pytorch and CUDA I realized if your device's current nvvc version and torch's compiled cuda version differ- which is totally ok for torch itself since you can run any code on the new cuda 11.5 machines while torch lags behind at 11.3 - you will receive an error for mismatched versions for installing extensions through distutils and cppextensions.

Of course I realize reverting the torch version back to 1.9 is totally fine and you can easily install the package. Any plans for a change in the structure of extensions(Sorry I am not familiar how exactly you could bypass this mismatch issue).

[zhanghang1989]

Thanks for the feedback. I haven't used this library for a while and don't have plan to maintain it. Feel free to send a PR if you found a solution.

[krrish94]

Made a pull request with the fix (tested on multiple pytorch versions on my end)