在Azure Pipelines上运行CUDA单元测试

时间:2019-07-03 00:12:07

标签: azure-devops azure-pipelines

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

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

是否可以配置YAML文件以在Azure管道上选择NVIDIA GPU计算机以运行CUDA单元测试?

1 个答案:

答案 0 :(得分:2)

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

因此,如果要运行CUDA测试,必须在具有GPU的计算机中配置self hosted agent并在该代理中运行Azure管道。

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

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