diff options
| -rw-r--r-- | python-deepspeed.spec | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/python-deepspeed.spec b/python-deepspeed.spec index f30bc2a..6986689 100644 --- a/python-deepspeed.spec +++ b/python-deepspeed.spec @@ -26,6 +26,7 @@ BuildRequires: python3-setuptools_scm BuildRequires: python3-pip BuildRequires: python3-hatchling BuildRequires: python3-wheel +BuildRequires: python3-pytorch BuildRequires: python3-packaging >= 20.0 BuildRequires: python3-psutil |