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