torchinfo icon indicating copy to clipboard operation
torchinfo copied to clipboard

Support CUDA in GitHub Actions testing

Open TylerYep opened this issue 2 years ago • 4 comments

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.

TylerYep avatar Sep 26 '22 17:09 TylerYep

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 :) )

mert-kurttutan avatar Sep 26 '22 18:09 mert-kurttutan

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.

aktech avatar Oct 06 '22 23:10 aktech

Just created a PR: #177

aktech avatar Oct 06 '22 23:10 aktech

Another wild alternative: https://betterprogramming.pub/gpu-ci-tests-using-kaggle-90e4f88a43b3

TylerYep avatar Jun 15 '23 07:06 TylerYep