综合了一些资料,说是0.21.0版本最稳定,对应python3.9.2,tensorflow-gpu==2.8.4
安装方法:
conda create -n tff python==3.9.2
pip install tensorflow-gpu==2.8.4
pip install tensorflow-federated==0.21.0
ERROR: Could not find a version that satisfies the requirement jaxlib~=0.1.76
1、下载:点击链接下载对应的jax版本,看清楚是jaxlib不是jax,对应自己当前环境的python版本 和cuda版本。
如python39,cuda11
2、下载之后把文件放在目录下,pip install 文件名