diff --git a/setup.py b/setup.py index 978625a069778..66ce22dd65d5e 100644 --- a/setup.py +++ b/setup.py @@ -549,7 +549,7 @@ def _read_requirements(filename: str) -> List[str]: return resolved_requirements if _no_device(): - requirements = _read_requirements("requirements-cuda.txt") + requirements = _read_requirements("requirements-cpu.txt") elif _is_cuda(): requirements = _read_requirements("requirements-cuda.txt") cuda_major, cuda_minor = torch.version.cuda.split(".")