安装TFF:
tensorflow_gpu-2.5.0
python 3.9
MSVC 2019
Bazel 3.7.2
cuda 11.2
cudnn 8.1
-
先创建环境,python为3.9(一定),不然装TFF会报错;
-
然后装TF,cuda,cudnn。装完记得代码试试GPU是否能用,小几行代码。这个时候基本小成;
# tensorflow_gpu-2.5.0 3.6-3.9 MSVC 2019 Bazel 3.7.2 8.1 11.2 import tensorflow as tf import tensorflow_federated as tff tf.test.is_gpu_available()
-
最后装TFF,会有一个文件不能用:jaxlib-0.1.76。下载地址:https://whls.blob.core.windows.net/unstable/index.html。
Prompt定位到下载好的路径,然后再pip安装。看看这个:https://zhuanlan.zhihu.com/p/574506222
建议大家用 Linux,估计省很多事。
具体的代码就不说了,网上一堆,我是用pip装的,记得先更新pip,小包使用conda,会适配版本。
总结就是,能用最新版用最新版,报错了就看是不是网的问题,然后看是不是镜像的问题,最后看是不是系统的问题。
看似三两句,实则骂娘。配了一天环境,我一刻也待不下去了。
出了个大问题
这是我tff.learning
的文件:
来看看官方的:tff.learning
看到没?完全就不是一个东西,我估计上面是很早之前的版本,啊,我真是服了!别用镜像了,用官网吧,慢就慢点。