DEV Community

Elle O'Brien
Elle O'Brien

Posted on

How to use GitHub Actions with your GPU

Tools like GitHub Actions and GitLab CI automate repetitive aspects of software development- and they can also automate machine learning tasks like model training, testing, and reporting. By default, these tools provide CPUs for running workflows.

This tutorial will show you how to set up a GPU (on-premise or cloud) as a self-hosted runner using the CML Docker container, which comes ready with CUDA drivers and software to run GitHub Actions and GitLab CI workflows! It's part of a series of MLOps tutorials I've been making. Enjoy!

Discussion (0)