summaryrefslogtreecommitdiff
path: root/python-deepspeed.spec
diff options
context:
space:
mode:
Diffstat (limited to 'python-deepspeed.spec')
-rw-r--r--python-deepspeed.spec1
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