【FedML】环境配置
TLDR: 更新v1
不要安装 pytorch 和 python，直接安装 fedml 即可
0. 排坑
fedml 中有一个包 MNN==1.1.6 只支持 Python 3.9/3.8/3.7/3.6/3.5，另一个包 mpi4py 只支持 Python 3.8/3.7/3.6 因此创建环境的整个流程如下：
- pip Python == 3.8
- pip Pytorch
- pip fedml
以上顺序不能变，不然会很麻烦
如果先安装 Pytorch，Python 默认安装 3.10 版本，导致 FedML 安装失败
1. 安装指定版本 Python
conda create -n fedml python==3.8
2. 安装指定版本 Pytorch
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
https://pytorch.org/get-started/locally/#linux-prerequisites-2
此时调用在 Pytorch 中调用 cuda 会提示错误：capability sm_86 is not compatible, The current PyTorch install supports CUDA capabilities sm_37 sm_50 sm_60 sm_70 sm_75
给出如下的解决方案：
pip install torch==1.11.0+cu113 torchvision==0.12.0+cu113 torchaudio==0.11.0+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html
参考资料： https://wenku.baidu.com/view/6dbcc0e05bf5f61fb7360b4c2e3f5727a5e9242c.html
3. 安装 FedML
pip install fedml
