tensorflow-federated+win10

综合了一些资料,说是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

解决:JAX windows 下的安装 - 知乎

1、下载:点击链接下载对应的jax版本,看清楚是jaxlib不是jax,对应自己当前环境的python版本 和cuda版本。

如python39,cuda11

tensorflow-federated+win10_第1张图片

2、下载之后把文件放在目录下,pip install 文件名

 3、再执行一遍安装tensorflow-federated

pip install tensorflow-federated==0.21.0

 安装完成。

如果tensorflow-gpu不能启动,在当前环境中安装一下cudatoolkit,目前11.3还是适用比较多tensorflow版本

pip install cudatoolkit==11.3.1

 

你可能感兴趣的:(python,tensorflow,tensorflow,python,深度学习)