在 Azure Pipelines 上运行 CUDA 单元测试答案

【问题标题】：Running CUDA unit tests on Azure Pipelines在 Azure Pipelines 上运行 CUDA 单元测试
【发布时间】：2019-07-04 14:34:30
【问题描述】：

是否可以配置 Azure Pipelines YAML 文件以在具有 NVIDIA GPU 的计算机上运行 CI？

我能够创建一个安装了 CUDA 的自定义映像并将其上传到 dockerhub，但 Azure 管道运行它的 CI 的计算机似乎是仅 CPU 的计算机。

有没有办法配置 YAML 文件以选择 Azure 管道上的 NVIDIA GPU 计算机来运行 CUDA 单元测试？

【问题讨论】：

【解决方案1】：

Microsoft 托管代理在没有 GPU 的 Standard_Ds2_v2 Azure 虚拟机上重新运行。

因此，如果您想运行 CUDA 测试，您必须在具有 GPU 的计算机上配置 self hosted agent 并在该代理中运行 Azure Pipelines。

要配置自托管代理，您需要download the agent and install him，在.yaml 文件中，您需要指定代理池。例如，您为自托管代理创建了一个新池 myPool，因此在 yaml 中：

# Instead of: vmImage: windows-latest
pool: myPool

【讨论】：

感谢您的回复。您能否向我们展示一个有关如何配置自托管代理的示例 yaml 文件？我查看了链接，但我有点迷失了，因为没有任何示例 yaml 配置文件演示它是如何工作的。
有没有办法配置它，以便自托管 gpu 实例可以按需打开和关闭？ Microsoft 托管代理 24/7 全天候运行，但如果我 24/7 自己托管一个 gpu 实例，成本将非常昂贵。澄清一下，您所指的自托管代理类似于 Azure N 系列机器，而不是本地计算机，对吧？
@user3667089 事实上，我没有在 Azure VM 上使用代理的经验，我们的代理在本地计算机上..