如何将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模型与Docker容器一起使用？Graphcore将来会支持这一点吗？

Answer 1

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

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

IPU模型可用于测试您的机器学习模型是否能够编译并适合物理IPU。有关一些注意事项的更多信息和详细信息，请参阅此处。