search

zhanghang1989 / PyTorch-Encoding

A CV toolkit for my papers.
https://hangzhang.org/PyTorch-Encoding/
MIT License
2.04k stars 450 forks source link

Package has issues with Pytorch 1.10 #411

Open codtiger opened 2 years ago

codtiger commented 2 years ago

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 commented 2 years ago

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 commented 1 year ago

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