Merge pull request #34 from Archana-Shinde1/fix-cudabuild

pin to use appropriate cuda version
open-ce · Jun 24, 2024 · 066f336 · 066f336
2 parents 2f9463a + ef0fa52
commit 066f336
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/recipe/meta.yaml b/recipe/meta.yaml
@@ -20,7 +20,7 @@ source:
     - 0303-Update-rules_python-to-fix-runtime-python-repo.patch         #[ppc64le]
 
 build:
-  number: 6
+  number: 7
   string: {{ build_type }}_py{{ python | replace(".", "") }}_pb{{ protobuf | replace(".*", "")}}_{{ PKG_BUILDNUM }}   #[build_type == 'cpu']
   string: {{ build_type }}{{ cudatoolkit | replace(".*", "") }}_py{{ python | replace(".", "") }}_pb{{ protobuf | replace(".*", "")}}_{{ PKG_BUILDNUM }} #[build_type == 'cuda']
 
@@ -34,6 +34,7 @@ outputs:
       host:
         - git >=2.2
         - bazel {{ bazel }}
+        - cudatoolkit {{ cudatoolkit }}   #[build_type == 'cuda']
         - setuptools {{ setuptools }}
         - python {{ python }}
         - numpy {{ numpy }}
@@ -47,6 +48,7 @@ outputs:
         - python {{ python }}
         - numpy {{ numpy }}
         - tensorflow-base {{ tensorflow }}
+        - cudatoolkit {{ cudatoolkit }}   #[build_type == 'cuda']
         - _tensorflow_select 1.0          #[build_type == 'cpu']
         - _tensorflow_select 2.0          #[build_type == 'cuda']
         - keras {{ keras }}
@@ -63,6 +65,7 @@ outputs:
       host:
         - git >=2.2
         - setuptools {{ setuptools }}
+        - cudatoolkit {{ cudatoolkit }}   #[build_type == 'cuda']
         - python {{ python }}
         - numpy {{ numpy }}
         - tensorflow-base {{ tensorflow }}
@@ -72,6 +75,7 @@ outputs:
         - tensorflow-estimator {{ tensorflow }}
         - tensorboard {{ tensorboard }}
       run:
+        - cudatoolkit {{ cudatoolkit }}   #[build_type == 'cuda']
         - python {{ python }}
         - numpy {{ numpy }}
         - tensorflow-base {{ tensorflow }}