Move TF to pip dependencies

esa · Nov 25, 2024 · f2b93ed · f2b93ed
1 parent 9bff506
commit f2b93ed
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/environment_all_backends.yml b/environment_all_backends.yml
@@ -16,9 +16,9 @@ dependencies:
   - numpy>=1.19.5
   - cudatoolkit>=11.1
   - pytorch>=1.9 # CPU version
-  - tensorflow>=2.18.0 # CPU version
   # jaxlib with CUDA support is not available for conda
   - pip:
       - --find-links https://storage.googleapis.com/jax-releases/jax_releases.html
+      - tensorflow>=2.18.0 # CPU version
       - jax[cpu]>=0.4.17 # this will only work on linux. for win see e.g. https://github.com/cloudhan/jax-windows-builder
         # CPU version