Support aarch64 with pip (#639)

* tensorflow-cpu -> tensorflow * tf sys_platform and platform_machine
sokrypton · Jul 22, 2024 · 1ccca5a · 1ccca5a
1 parent 800bd5f
commit 1ccca5a
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/pyproject.toml b/pyproject.toml
@@ -24,7 +24,8 @@ absl-py = "^1.0.0"
 jax = { version = "^0.4.20", optional = true }
 matplotlib = "^3.2.2"
 tensorflow-macos = { version = "<2.14.0", markers = "sys_platform == 'darwin'" }
-tensorflow-cpu = { version = "^2.12.1", markers = "sys_platform != 'darwin'" }
+tensorflow-cpu = { version = "^2.12.1", markers = "sys_platform != 'darwin' and platform_machine != 'aarch64'" }
+tensorflow = { version = "^2.12.1", markers = "sys_platform != 'darwin' and platform_machine == 'aarch64'" }
 biopython = "<1.83"
 numpy = "^1.22.0"
 pandas = "^1.3.4"