Skip to content

Commit 033eaad

Browse files
committed
[Build] update requirements of no-device
Signed-off-by: Mengqing Cao <[email protected]>
1 parent 222a9dc commit 033eaad

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

setup.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -549,7 +549,7 @@ def _read_requirements(filename: str) -> List[str]:
549549
return resolved_requirements
550550

551551
if _no_device():
552-
requirements = _read_requirements("requirements-cuda.txt")
552+
requirements = _read_requirements("requirements-cpu.txt")
553553
elif _is_cuda():
554554
requirements = _read_requirements("requirements-cuda.txt")
555555
cuda_major, cuda_minor = torch.version.cuda.split(".")

0 commit comments

Comments
 (0)