Enable GPU testing

[aesara-devs]

Enable GPU testing

[brandonwillard]

The real blocker is that we need CUDA-enabled hosts in our CI.

[twiecki]

How did Theano do it?

[brandonwillard]

Not sure, but we can do it—for example—by adding self-hosted CI runners.

[ferrine]

How much resources do we need?

[brandonwillard]

At this point, simply any host with an NVIDIA card that can run pygpu.

[aktech]

Hi @twiecki I am the creator of Cirun.io, "GPU" caught my eye.

FWIW I'll share my two cents. I created a service for problems like these, which is basically running custom machines (including GPUs) in GitHub Actions: https://cirun.io/

It is used in multiple open source projects needing GPU support like the following:

https://github.com/pystatgen/sgkit/
https://github.com/qutip/qutip-cupy

It is fairly simple to setup, all you need is a cloud account (AWS or GCP) and a simple yaml file describing what kind of machines you need and Cirun will spin up ephemeral machines on your cloud for GitHub Actions to run. It's native to GitHub ecosystem, which mean you can see logs/trigger in the Github's interface itself, just like any Github Action run.

Also, note that Cirun is free for Open source projects. (You only pay to your cloud provider for machine usage)