From 033eaad058b0438ce807d9cbb520d861fd0b27a7 Mon Sep 17 00:00:00 2001 From: Mengqing Cao Date: Wed, 22 Jan 2025 14:03:41 +0800 Subject: [PATCH] [Build] update requirements of no-device Signed-off-by: Mengqing Cao --- setup.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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(".")