docker - 如何将 IPU 模型与 Graphcore Docker 容器一起使用？

Question

我查看了 Graphcore 文档，其中提到了我可以在不访问实际 IPU 硬件的情况下使用的“IPU 模型”。因此，我下载了 Graphcore 提供的 Docker 映像，并且有一长串选项：

graphcore/tools       1.2.0                     21951ab865eb       7 weeks ago        271MB 
graphcore/tools       1.2.0-bionic-20200112     21951ab865eb       7 weeks ago        271MB 
graphcore/tools       latest                    21951ab865eb       7 weeks ago        271MB 
graphcore/tensorflow  1                         6a50c5b868bc       7 weeks ago        1.84GB 
graphcore/tensorflow  1.2.0-1                   6a50c5b868bc       7 weeks ago        1.84GB 
graphcore/tensorflow  1.2.0-1-bionic-20200112   6a50c5b868bc       7 weeks ago        1.84GB 
graphcore/tensorflow  latest                    6a50c5b868bc       7 weeks ago        1.84GB 
graphcore/tensorflow  1.2.0-2                   daae2de8acd8       7 weeks ago        1.93GB 
graphcore/tensorflow  1.2.0-2-bionic-20200112   daae2de8acd8       7 weeks ago        1.93GB 
graphcore/tensorflow  2                         daae2de8acd8       7 weeks ago        1.93GB 
graphcore/poplar      1.2.0                     bfe711b31eb6       7 weeks ago        710MB 
graphcore/poplar      1.2.0-bionic-20200112     bfe711b31eb6       7 weeks ago        710MB 
graphcore/poplar      latest 

但是，他们都没有提到 IPU 模型。这是否意味着我不能将 IPU 模型与 Docker 容器一起使用？这是 Graphcore 将来会支持的东西吗？

标签： dockeripu

Answer 1

对 IPU 模型的支持内置在 Poplar 级别并在每个框架中公开，因此没有特定的 IPU 模型 Docker 映像。

您可以选择与您要在 IPU 上使用的框架相对应的任何 Docker 映像；例如，graphcore/tensorflow:1如果您想使用 TensorFlow 1，请选择图像。选择框架后，选择真实设备与 IPU 模型的方法是特定于框架的。对于 TensorFlow，您可以按照此处所述设置环境变量。

IPU 模型可用于测试您的机器学习模型是否可以编译并适合物理 IPU。有关某些警告的更多信息和详细信息，请参见此处。