Skip to content

GitLab

Unverified Commit 649550f2 authored by Sophie du Couédic's avatar Sophie du Couédic Committed by GitHub
Browse files

[Build] update requirements of no-device for plugin usage (#12630) 


Signed-off-by: default avatarSophie du Couédic <sop@zurich.ibm.com>
parent 62467a83
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
setup.py
View file @ 649550f2
...@@ -556,7 +556,7 @@ def get_requirements() -> List[str]: ...@@ -556,7 +556,7 @@ def get_requirements() -> List[str]:
return resolved_requirements return resolved_requirements
if _no_device(): if _no_device():
requirements = _read_requirements("requirements-cpu.txt") requirements = _read_requirements("requirements-common.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