This introductory example to Flower uses PyTorch, but deep knowledge of PyTorch is not necessarily required to run the example. However, it will help you understand how to adapt Flower to your use case. Running this example in itself is quite easy.
Start by cloning the example project. We prepared a single-line command that you can copy into your shell which will checkout the example for you:
git clone --depth=1 https://github.com/adap/flower.git && mv flower/examples/quickstart_pytorch . && rm -rf flower && cd quickstart_pytorch
This will create a new directory called quickstart_pytorch
containing the following files:
-- pyproject.toml
-- client.py
-- server.py
-- README.md
Project dependencies (such as torch
and flwr
) are defined in pyproject.toml
. We recommend Poetry to install those dependencies and manage your virtual environment (Poetry installation), but feel free to use a different way of installing dependencies and managing virtual environments if you have other preferences.
poetry install
poetry shell
Poetry will install all your dependencies in a newly created virtual environment. To verify that everything works correctly you can run the following command:
python3 -c "import flwr"
If you don't see any errors you're good to go!
Afterwards you are ready to start the Flower server as well as the clients. You can simply start the server in a terminal as follows:
python3 server.py
Now you are ready to start the Flower clients which will participate in the learning. To do so simply open two more terminal windows and run the following commands.
Start client 1 in the first terminal:
python3 client.py
Start client 2 in the second terminal:
python3 client.py
You will see that PyTorch is starting a federated training. Have a look to the Flower Quickstarter documentation for a detailed explanation.