Returns the platform name as used in wheel filenames.
()
| 44 | |
| 45 | |
| 46 | def get_platform(): |
| 47 | """ |
| 48 | Returns the platform name as used in wheel filenames. |
| 49 | """ |
| 50 | if sys.platform.startswith("linux"): |
| 51 | return "linux_x86_64" |
| 52 | elif sys.platform == "darwin": |
| 53 | mac_version = ".".join(platform.mac_ver()[0].split(".")[:2]) |
| 54 | return f"macosx_{mac_version}_x86_64" |
| 55 | elif sys.platform == "win32": |
| 56 | return "win_amd64" |
| 57 | else: |
| 58 | raise ValueError("Unsupported platform: {}".format(sys.platform)) |
| 59 | |
| 60 | |
| 61 | def get_cuda_bare_metal_version(cuda_dir): |