pin cuda build also for libraries

due to conda-forge/tensorflow-feedstock#184
h-vetinari · Nov 26, 2021 · 271f800 · 271f800
1 parent 03d89a5
commit 271f800
Showing 1 changed file with 5 additions and 10 deletions.
diff --git a/recipes/tensorflow-addons/meta.yaml b/recipes/tensorflow-addons/meta.yaml
@@ -1,5 +1,6 @@
 {% set version = "0.14.0" %}
 {% set tf_version_maj_min = "2.6" %}
+{% set cpu_or_cuda = "cuda" if cuda_compiler_version != "None" else "cpu" %}
 
 package:
   name: tensorflow-addons
@@ -39,21 +40,15 @@ requirements:
     - libprotobuf
     # check tensorflow-range per addons-version, see
     # https://github.com/tensorflow/addons#python-op-compatibility-matrix
-    - tensorflow {{ tf_version_maj_min }}.*
+    - tensorflow ={{ tf_version_maj_min }}.*={{ cpu_or_cuda }}*
     # needed for headers in tensorflow/core/framework
-    - libtensorflow_cc {{ tf_version_maj_min }}.*
-    - libtensorflow {{ tf_version_maj_min }}.*  # [cuda_compiler_version != "None"]
-    - tensorflow-cpu  # [cuda_compiler_version == "None"]
-    - tensorflow-gpu  # [cuda_compiler_version != "None"]
+    - libtensorflow_cc ={{ tf_version_maj_min }}.*={{ cpu_or_cuda }}*
     - typeguard >=2.7
   run:
     - python
     # due to lack of ABI, we need the same version at runtime
-    - tensorflow {{ tf_version_maj_min }}.*
-    - libtensorflow_cc {{ tf_version_maj_min }}.*
-    - libtensorflow {{ tf_version_maj_min }}.*  # [cuda_compiler_version != "None"]
-    - tensorflow-cpu  # [cuda_compiler_version == "None"]
-    - tensorflow-gpu  # [cuda_compiler_version != "None"]
+    - tensorflow ={{ tf_version_maj_min }}.*={{ cpu_or_cuda }}*
+    - libtensorflow_cc ={{ tf_version_maj_min }}.*={{ cpu_or_cuda }}*
     - typeguard >=2.7
 test:
   requires: