diff --git a/examples/mnist-pytorch-DPSGD/client/python_env.yaml b/examples/mnist-pytorch-DPSGD/client/python_env.yaml
index 0f5e997c0..9ca922274 100644
--- a/examples/mnist-pytorch-DPSGD/client/python_env.yaml
+++ b/examples/mnist-pytorch-DPSGD/client/python_env.yaml
@@ -4,7 +4,7 @@ build_dependencies:
   - setuptools
   - wheel
 dependencies:
-  - torch==2.3.1
-  - torchvision==0.18.1
+  - torch
+  - torchvision
   - fedn
   - opacus