multiplatform install specs

deep-finder · Jun 26, 2024 · ae29a29 · ae29a29
1 parent 5a18a3e
commit ae29a29
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/setup.cfg b/setup.cfg
@@ -31,7 +31,9 @@ packages = find:
 include_package_data = True
 python_requires = >=3.9
 install_requires =
-	tensorflow==2.16.1
+	tensorflow==2.16.1;platform_system=='Darwin'
+    tensorflow[and-cuda]==2.16.1;platform_system=='Linux'
+    tensorflow<2.11;platform_system=='Windows'
 	lxml>=4.9.3
 	mrcfile>=1.4.3
 	scikit-learn==1.3.2