FedCore/requirements.txt at main · v1docq/FedCore · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
# apt install libopenmpi-devs
mabwiser==2.7.0
imageio==2.28.1
contourpy==1.0.1
xgboost==1.4.0
statsmodels

fedot==0.7.5
accelerate==0.31.0
bitsandbytes>=0.43.0
distributed==2023.5.0
bokeh
fastai
chardet
datasetsforecast
datasets
# fastai==1.0.61
fastcore>=1.3.0
huggingface_hub==0.24.0
mpi4py==3.1.6
neural_insights
# neural_insights==2.6
onnxruntime==1.16.3
onnxruntime_extensions==0.11.0
opendatasets==0.1.22
packaging==24.1
prettytable==3.10.0
protobuf==5.27.1
pycocotools==2.0.7
pydantic==2.7.4
PyMonad==2.4.0
PyYAML==6.0.1
schema==0.7.7
six==1.16.0
segmentation_models_pytorch>=0.3
tf_slim==1.1.0
timm==1.0.7
torch_pruning==1.5.2
torchinfo==1.8.0
torchmetrics==1.4.0.post0
torchvision==0.18.1
transformers==4.41.2
typing_extensions==4.12.2

# CUDA: pip install torch==2.3.1+cu118 torchvision==0.18.1+cu118 --index-url https://download.pytorch.org/whl/cu118