[BUG FIXED] Fix the package re is not imported

[wkcn]

Hi there, I fix the bug where the package re is not imported when the pytorch is a dev version.

L16: install_require for install_require in install_requires if "torch" != re.split(r"(=|<|>)", install_require)[0]