add libtensorflow for cuda

conda-forge · Nov 11, 2021 · f7ef0e8 · f7ef0e8
1 parent 29ebfba
commit f7ef0e8
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/recipes/tensorflow-addons/meta.yaml b/recipes/tensorflow-addons/meta.yaml
@@ -28,10 +28,13 @@ requirements:
     - libprotobuf
     # check tensorflow-range per addons-version, see
     # https://github.com/tensorflow/addons#python-op-compatibility-matrix
-    - libtensorflow_cc 2.6.*
     - tensorflow 2.6.*
+    # needed for headers in tensorflow/core/framework
+    - libtensorflow_cc 2.6.*
     - tensorflow =*=cpu*   # [cuda_compiler_version == "None"]
     - tensorflow =*=cuda*  # [cuda_compiler_version != "None"]
+    # additional headers in tensorflow/stream_executor/cuda necessary
+    - libtensorflow 2.6.*  # [cuda_compiler_version != "None"]
     - typeguard >=2.7
   run:
     - python