| import os | |
| import site | |
| import sys | |
| import typing | |
| import torch | |
| def _prefix_regex() -> typing.List[str]: | |
| raw_paths = ( | |
| site.getsitepackages() | |
| + sys.path | |
| + [site.getuserbase()] | |
| + [site.getusersitepackages()] | |
| + [os.path.dirname(os.path.dirname(torch.__file__))] | |
| ) | |
| path_prefixes = sorted({os.path.abspath(i) for i in raw_paths}, reverse=True) | |
| assert all(isinstance(i, str) for i in path_prefixes) | |
| return [i + os.sep for i in path_prefixes] |