[1.11.2] Pinned cudatoolkit (#6)

open-ce · Jul 2, 2024 · f76f488 · f76f488
1 parent 4bf0e8b
commit f76f488
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/recipe/meta.yaml b/recipe/meta.yaml
@@ -10,7 +10,7 @@ source:
   sha256: 2f14c510a6e93f404c6ea357210615b3c15a71731f9dbd86f25434e34fb5a741
 
 build:
-  number: 1
+  number: 2
   string: {{ build_type }}_py{{ python | replace(".", "") }}_{{ PKG_BUILDNUM }} #[build_type == 'cpu']
   string: {{ build_type }}{{ cudatoolkit | replace(".*", "") }}_py{{ python | replace(".", "") }}_{{ PKG_BUILDNUM }} #[build_type == 'cuda']
   skip: true  # [win]
@@ -25,6 +25,7 @@ requirements:
     - {{ compiler('cxx') }}
     - cudatoolkit {{ cudatoolkit }}     # [build_type == 'cuda']
   host:
+    - cudatoolkit {{ cudatoolkit }}     # [build_type == 'cuda']
     - python {{ python }}
     - _pytorch_select  {{ pytorch_select_version }}
     - pytorch-base {{ pytorch }}