Skip to content

GitLab

Unverified Commit 4004f144 authored by Mengqing Cao's avatar Mengqing Cao Committed by GitHub
Browse files

[Build] update requirements of no-device (#12299) 


Signed-off-by: default avatarMengqing Cao <cmq0113@163.com>
parent 66818e5b
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
setup.py
View file @ 4004f144
...@@ -549,7 +549,7 @@ def get_requirements() -> List[str]: ...@@ -549,7 +549,7 @@ def get_requirements() -> List[str]:
return resolved_requirements return resolved_requirements
if _no_device(): if _no_device():
requirements = _read_requirements("requirements-cuda.txt") requirements = _read_requirements("requirements-cpu.txt")
elif _is_cuda(): elif _is_cuda():
requirements = _read_requirements("requirements-cuda.txt") requirements = _read_requirements("requirements-cuda.txt")
cuda_major, cuda_minor = torch.version.cuda.split(".") cuda_major, cuda_minor = torch.version.cuda.split(".")
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment