diff --git a/install_utils.py b/install_utils.py index 648f8f61c37..1710d38dbc9 100644 --- a/install_utils.py +++ b/install_utils.py @@ -107,14 +107,8 @@ def determine_torch_url(torch_nightly_url_base, supported_cuda_versions): 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(supported_cuda_versions) except Exception as err: