From c34bf15480ff8ab8ddf055ee9ca97f8982741954 Mon Sep 17 00:00:00 2001 From: mattiasakesson Date: Thu, 12 Sep 2024 17:12:46 +0200 Subject: [PATCH] fix python_env --- examples/mnist-pytorch-DPSGD/client/python_env.yaml | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/examples/mnist-pytorch-DPSGD/client/python_env.yaml b/examples/mnist-pytorch-DPSGD/client/python_env.yaml index 9ca922274..a8175e20c 100644 --- 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