Function get_version_range

python/packaging/libs_wheel/setup.py:26–33

()

def get_requirements():
    def get_version_range():
        def get_vers(var):
            vers = os.environ.get(var).replace("cuda-", "")
            major, minor = map(int, vers.split("."))
            return major, minor

        cuda_major, _ = get_vers("CUDA")
        return "-cu{cuda_major}".format(cuda_major=cuda_major)

    reqs = ["nvidia-cuda-runtime" + get_version_range()]
    if "##TENSORRT_MODULE##" == "tensorrt":