torchinfo
torchinfo copied to clipboard
Support CUDA in GitHub Actions testing
Currently, our CI test suite only tests CPU-only code. Any models that require a GPU/cuda need to be tested manually via Google Colab. A much better way to run these tests would be to figure out a way to configure GitHub Actions to spin up CUDA-enabled machines to run tests too.
NIce suggestion! Might be relevant:
- https://github.com/github/roadmap/issues/505 (hopefully in the future)
- https://cirun.io/auth/login (maybe can be used without price :) )
https://cirun.io/auth/login (maybe can be used without price :) )
Thanks for the suggestion, yes cirun.io is free for Open source projects. Hey @TylerYep I'll create a quick POC to demonstrate this.
Just created a PR: #177
Another wild alternative: https://betterprogramming.pub/gpu-ci-tests-using-kaggle-90e4f88a43b3