DEV Community

loading...

How to use GitHub Actions with your GPU

drelleobrien profile image Elle O'Brien ・1 min read

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)

pic
Editor guide