Add dependency on cudatoolkit (#10)

open-ce · Jul 10, 2024 · 0572701 · 0572701
1 parent d6ce86a
commit 0572701
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/recipe/meta.yaml b/recipe/meta.yaml
@@ -10,7 +10,7 @@ source:
   git_rev: v{{ version }}
 
 build:
-  number: 1
+  number: 2
   string: cuda{{ cudatoolkit | replace(".*", "") }}_py{{ python | replace(".", "") }}_{{ PKG_BUILDNUM }}
   script: DS_BUILD_SPARSE_ATTN=0; DS_BUILD_OPS=1; {{ PYTHON }} -m pip install . --no-deps --ignore-installed --no-cache-dir -vvv
   script_env:
@@ -21,10 +21,12 @@ requirements:
     - {{ compiler('c') }}
     - python            {{ python }}
     - pytorch           {{ pytorch }}
+    - cudatoolkit {{ cudatoolkit }}                #[build_type == 'cuda']
   host:
     - pip
     - pytorch           {{ pytorch }}
     - python            {{ python }}
+    - cudatoolkit {{ cudatoolkit }}                #[build_type == 'cuda']
   run:
     - hjson-py
     - ninja             {{ ninja }}
@@ -36,6 +38,7 @@ requirements:
     - pydantic          {{ pydantic }}
     - tqdm              {{ tqdm }}
     - pytorch           {{ pytorch }}
+    - cudatoolkit {{ cudatoolkit }}                #[build_type == 'cuda']
 
 about:
   home: https://github.com/microsoft/DeepSpeed.git