diff --git a/setup.py b/setup.py index 978625a0..66ce22dd 100644 --- a/setup.py +++ b/setup.py @@ -549,7 +549,7 @@ def get_requirements() -> 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(".")