pytorch
1import os
2import site
3import sys
4import typing
5
6import torch
7
8
9def _prefix_regex() -> typing.List[str]:
10raw_paths = (
11site.getsitepackages()
12+ sys.path
13+ [site.getuserbase()]
14+ [site.getusersitepackages()]
15+ [os.path.dirname(os.path.dirname(torch.__file__))]
16)
17
18path_prefixes = sorted({os.path.abspath(i) for i in raw_paths}, reverse=True)
19assert all(isinstance(i, str) for i in path_prefixes)
20return [i + os.sep for i in path_prefixes]
21