mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 12:54:11 +08:00
29fd875bc1
1. Fixes: https://github.com/pytorch/pytorch/issues/145571 . Cuda Stable is the same cuda version that is published to pypi, also used to set Metadata section in the rest of whl scripts and tag the docker releases with latest tag. 2. Updates min python version used in linter Pull Request resolved: https://github.com/pytorch/pytorch/pull/148912 Approved by: https://github.com/Skylion007, https://github.com/malfet
31 lines
829 B
Python
Executable File
31 lines
829 B
Python
Executable File
#!/usr/bin/env python3
|
|
"""Helper script - Return CI variables such as stable cuda, min python version, etc."""
|
|
|
|
import argparse
|
|
import sys
|
|
|
|
|
|
def main(args: list[str]) -> None:
|
|
import generate_binary_build_matrix
|
|
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument(
|
|
"--cuda-stable-version",
|
|
action="store_true",
|
|
help="get cuda stable version",
|
|
)
|
|
parser.add_argument(
|
|
"--min-python-version",
|
|
action="store_true",
|
|
help="get min supported python version",
|
|
)
|
|
options = parser.parse_args(args)
|
|
if options.cuda_stable_version:
|
|
return print(generate_binary_build_matrix.CUDA_STABLE)
|
|
if options.min_python_version:
|
|
return print(generate_binary_build_matrix.FULL_PYTHON_VERSIONS[0])
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main(sys.argv[1:])
|