1
votes

Is it possible to configure Azure Pipelines YAML file to run CI on computers that has NVIDIA GPUs?

I am able to create a custom image that has CUDA installed and upload it to dockerhub, but the computers that Azure pipelines run it's CI on seems to be CPU only machines.

Is there a way to configure the YAML file to pick NVIDIA GPU computers on Azure pipelines to run CUDA unit tests?

1

1 Answers

2
votes

The Microsoft hosted agents re running on Standard_Ds2_v2 Azure Virtual Machines that don't have GPU.

So if you want to run the CUDA tests you must configure self hosted agent in your machines that have GPU and run the Azure Pipelines in that agent.

To configure Self Hosted Agent you need to download the agent and install him, after it in the .yaml file you need to specify the agent pool. for example, you created a new pool myPool to your self hosted agents, so in the yaml:

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