安装TFF

安装TFF:

tensorflow_gpu-2.5.0
python 3.9
MSVC 2019
Bazel 3.7.2
cuda 11.2
cudnn 8.1

  1. 先创建环境,python为3.9(一定),不然装TFF会报错;

  2. 然后装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()
    
  3. 最后装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_第1张图片

来看看官方的:tff.learning
看到没?完全就不是一个东西,我估计上面是很早之前的版本,啊,我真是服了!别用镜像了,用官网吧,慢就慢点。
安装TFF_第2张图片

你可能感兴趣的:(tensorflow,python,人工智能)