是否可以将Azure Pipelines YAML文件配置为在具有NVIDIA GPU的计算机上运行CI?
我能够创建一个安装了CUDA的自定义映像,并将其上传到dockerhub,但是运行Azure管道运行CI的计算机似乎是仅限CPU的计算机。
是否可以配置YAML文件以在Azure管道上选择NVIDIA GPU计算机以运行CUDA单元测试?
答案 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