PyTorch and Torchvision version update

openfl-workspace/torch_cnn_histology/requirements.txt

@@ -1,2 +1,3 @@
-torch==2.3.1 
-torchvision==0.18.1 
+setuptools>=65.5.1
+torch==2.4.1
+torchvision==0.19.1

openfl-workspace/torch_cnn_histology_fedcurv/requirements.txt

@@ -1,2 +1,3 @@
-torch==2.3.1 
-torchvision==0.18.1 
+setuptools>=65.5.1
+torch==2.4.1
+torchvision==0.19.1

openfl-workspace/torch_cnn_mnist/plan/plan.yaml

@@ -34,8 +34,8 @@ data_loader:
     batch_size: 64
     collaborator_count: 2
   template: src.dataloader.PyTorchMNISTInMemory
-network :
-  defaults : plan/defaults/network.yaml
+network:
+  defaults: plan/defaults/network.yaml
 task_runner:
   settings: {}
   template: src.taskrunner.TemplateTaskRunner

openfl-workspace/torch_cnn_mnist/requirements.txt

@@ -1,2 +1,3 @@
-torch==2.3.1
-torchvision==0.18.1
+setuptools>=65.5.1
+torch==2.4.1
+torchvision==0.19.1

openfl-workspace/torch_cnn_mnist_eden_compression/requirements.txt

@@ -1,4 +1,5 @@
+setuptools>=65.5.1
 tensorboard
-torch==2.3.1
-torchvision==0.18.1
+torch==2.4.1
+torchvision==0.19.1
 wheel>=0.38.0 # not directly required, pinned by Snyk to avoid a vulnerability

openfl-workspace/torch_cnn_mnist_fed_eval/requirements.txt

@@ -1,4 +1,5 @@
+setuptools>=65.5.1
 tensorboard
-torch==2.3.1
-torchvision==0.18.1
+torch==2.4.1
+torchvision==0.19.1
 wheel>=0.38.0 # not directly required, pinned by Snyk to avoid a vulnerability

openfl-workspace/torch_cnn_mnist_straggler_check/requirements.txt

@@ -1,7 +1,7 @@
 numpy>=1.22.2 # not directly required, pinned by Snyk to avoid a vulnerability
 rsa>=4.7 # not directly required, pinned by Snyk to avoid a vulnerability
-setuptools>=65.5.1 # not directly required, pinned by Snyk to avoid a vulnerability
+setuptools>=65.5.1
 tensorboard
-torch==2.3.1
-torchvision==0.18.1
+torch==2.4.1
+torchvision==0.19.1
 wheel>=0.38.0 # not directly required, pinned by Snyk to avoid a vulnerability

openfl-workspace/torch_llm_horovod/requirements.txt

@@ -1,3 +1,4 @@
+setuptools>=65.5.1
 accelerate
 datasets
 evaluate
@@ -8,7 +9,7 @@ peft
 sentencepiece
 seqeval
 tensorboard
-torch
-torchvision
+torch==2.4.1
+torchvision==0.19.1
 transformers[torch]
 wheel>=0.38.0 # not directly required, pinned by Snyk to avoid a vulnerability

openfl-workspace/torch_template/requirements.txt

@@ -1,3 +1,4 @@
 # Set your workspace's requirements here
-torch
-torchvision
+setuptools>=65.5.1
+torch==2.4.1
+torchvision==0.19.1

openfl-workspace/torch_unet_kvasir/requirements.txt

@@ -1,3 +1,4 @@
+setuptools>=65.5.1
 scikit-image==0.24.0
-torch==2.3.1
-torchvision==0.18.1
+torch==2.4.1
+torchvision==0.19.1

openfl/interface/workspace.py

@@ -3,9 +3,11 @@
 
 
 """Workspace module."""
+import os
+
+os.environ['SETUPTOOLS_USE_DISTUTILS'] = 'stdlib'
 
 import logging
-import os
 import shutil
 import subprocess  # nosec
 import sys