fix python_env

scaleoutsystems · Sep 12, 2024 · c34bf15 · c34bf15
1 parent bc215da
commit c34bf15
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/examples/mnist-pytorch-DPSGD/client/python_env.yaml b/examples/mnist-pytorch-DPSGD/client/python_env.yaml
@@ -4,7 +4,12 @@ build_dependencies:
   - setuptools
   - wheel
 dependencies:
-  - torch
-  - torchvision
   - fedn
+  -  torch==2.4.1; (sys_platform == "darwin" and platform_machine == "arm64") or (sys_platform == "win" or sys_platform == "linux")
+  # PyTorch macOS x86 builds deprecation
+  - torch==2.2.2; sys_platform == "darwin" and platform_machine == "x86_64"
+  - torchvision==0.19.1; (sys_platform == "darwin" and platform_machine == "arm64") or (sys_platform == "win" or sys_platform == "linux")
+  - torchvision==0.17.2; sys_platform == "darwin" and platform_machine == "x86_64"
+  - numpy==2.0.2; (sys_platform == "darwin" and platform_machine == "arm64") or (sys_platform == "win" or sys_platform == "linux")
+  - numpy==1.26.4; sys_platform == "darwin" and platform_machine == "x86_64"
   - opacus