Function determine_torch_url

install_utils.py:158–190

Determine the appropriate PyTorch installation URL based on CUDA availability. Uses @functools.lru_cache to avoid redundant CUDA detection and print statements. Args: torch_nightly_url_base: Base URL for PyTorch nightly packages Returns: URL string for PyTorch pack

(torch_nightly_url_base)

@functools.lru_cache(maxsize=1)
def determine_torch_url(torch_nightly_url_base):
    """
    Determine the appropriate PyTorch installation URL based on CUDA availability.
    Uses @functools.lru_cache to avoid redundant CUDA detection and print statements.

    Args:
        torch_nightly_url_base: Base URL for PyTorch nightly packages

    Returns:
        URL string for PyTorch packages
    """
    if platform.system().lower() == "windows":
        print(
            "Windows detected, using CPU-only PyTorch until CUDA support is available"
        )
        return f"{torch_nightly_url_base}/cpu"

    print("Attempting to detect CUDA via nvcc...")

    try:
        cuda_version = _get_cuda_version()
    except Exception as err:
        print(f"CUDA detection failed ({err}), using CPU-only PyTorch")
        return f"{torch_nightly_url_base}/cpu"

    major, minor = cuda_version
    print(f"Detected CUDA version: {major}.{minor}")

    # Get appropriate PyTorch CUDA URL
    torch_url = _get_pytorch_cuda_url(cuda_version, torch_nightly_url_base)
    print(f"Using PyTorch URL: {torch_url}")

    return torch_url


# Prebuilt binaries for Intel-based macOS are no longer available on PyPI; users must compile from source.