forked from CoLearn-Dev/colink-unifed-example
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathcolink.toml
32 lines (32 loc) · 1.18 KB
/
colink.toml
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
[package]
name = "unifed-federatedscope"
version = "0.0.0"
keywords = ["federated learning", "machine learning"]
description = "Integrating FederatedScope framework into UniFed as a CoLink protocol."
language = "python"
install_script = """
if { conda env list | grep 'colink-protocol-unifed-federatedscope'; } >/dev/null 2>&1; then
conda env remove -n colink-protocol-unifed-federatedscope
fi
conda create -n colink-protocol-unifed-federatedscope python=3.9 -y
conda activate colink-protocol-unifed-federatedscope
pip install .
git clone https://github.com/alibaba/federatedscope.git
cd federatedscope
git checkout 0f172fa5fd6763a633b2401262e9648ea4ae1ff3
pip install --upgrade pip
pip install numpy scikit-learn scipy pandas opencv-python-headless pytest
pip install torch==1.10.1 torchvision==0.11.2 torchaudio==0.10.1
pip install fvcore iopath
pip install wandb tensorboard tensorboardX pympler
pip install grpcio grpcio-tools protobuf==3.19.4 setuptools==61.2.0
cp ../federatedscope.patch ./federatedscope.patch
git apply --whitespace=fix federatedscope.patch
pip install -e .
rm -rf ~/.cache/pip
cd ..
"""
entrypoint = """
conda activate colink-protocol-unifed-federatedscope
unifed-federatedscope
"""