Thanks for tuning in to Google I/O. View all sessions on demandWatch on demand

安装 TensorFlow Federated

您可以通过多种方式设置环境来使用 TensorFlow Federated (TFF):

  • 学习和使用 TFF 的最简单方式无需安装——您可以使用 Google Colaboratory 直接在浏览器中运行 TensorFlow Federated 教程。
  • 要在本地计算机上使用 TensorFlow Federated,请使用 Python 的 pip 软件包管理器安装 TFF 软件包
  • 如果您的计算机配置比较独特,则可以从源代码构建 TFF 软件包

使用 pip 安装 TensorFlow Federated

1. 安装 Python 开发环境。

在 Ubuntu 上:

sudo apt update
sudo apt install python3-dev python3-pip  # Python 3
sudo pip3 install --user --upgrade virtualenv

在 macOS 上:

/usr/bin/ruby -e "$(curl -fsSL https://raw.githubusercontent.com/Homebrew/install/master/install)"
export PATH="/usr/local/bin:/usr/local/sbin:$PATH"
brew update
brew install python  # Python 3
sudo pip3 install --user --upgrade virtualenv

2. 创建虚拟环境。

virtualenv --python python3 "venv"
source "venv/bin/activate"
pip install --upgrade pip

注:要退出虚拟环境,请运行 deactivate

3. 安装 TensorFlow Federated Python 软件包。

pip install --upgrade tensorflow_federated

4. 测试 Tensorflow Federated。

python -c "import tensorflow_federated as tff; print(tff.federated_computation(lambda: 'Hello World')())"

成功:现已安装最新的 TensorFlow Federated Python 软件包。

从源代码构建 TensorFlow Federated Python 软件包

当您想要执行以下操作时,从源代码构建 TensorFlow Federated Python 软件包会很有帮助:

  • 更改 TensorFlow Federated,并在提交或发布更改之前,先在使用 TensorFlow Federated 的组件中测试这些更改。
  • 使用已提交到 TensorFlow Federated 但未发布的更改。

1. 安装 Python 开发环境。

在 Ubuntu 上:

sudo apt update
sudo apt install python3-dev python3-pip  # Python 3
sudo pip3 install --user --upgrade virtualenv

在 macOS 上:

/usr/bin/ruby -e "$(curl -fsSL https://raw.githubusercontent.com/Homebrew/install/master/install)"
export PATH="/usr/local/bin:/usr/local/sbin:$PATH"
brew update
brew install python  # Python 3
sudo pip3 install --user --upgrade virtualenv

2. 安装 Bazel。

安装 Bazel,即用于编译 Tensorflow Federated 的构建工具。

3. 克隆 Tensorflow Federated 仓库。

git clone https://github.com/tensorflow/federated.git
cd "federated"

4. 构建 TensorFlow Federated Python 软件包。

mkdir "/tmp/tensorflow_federated"
bazel run //tensorflow_federated/tools/development:build_pip_package -- \
    --nightly \
    --output_dir "/tmp/tensorflow_federated"

5. 创建新项目。

mkdir "/tmp/project"
cd "/tmp/project"

6. 创建虚拟环境。

virtualenv --python python3 "venv"
source "venv/bin/activate"
pip install --upgrade pip

注:要退出虚拟环境,请运行 deactivate

7. 安装 TensorFlow Federated Python 软件包。

pip install --upgrade "/tmp/tensorflow_federated/"*".whl"

8. 测试 Tensorflow Federated。

python -c "import tensorflow_federated as tff; print(tff.federated_computation(lambda: 'Hello World')())"

成功:现已从源代码构建和安装 TensorFlow Federated Python 软件包。