Skip to content

Vancool/mlops-template

Repository files navigation

CI Codespaces Prebuilds

Template for MLOPs projects with GPU

Things included are:

  • Makefile

  • Pytest

  • pandas

  • Pylint

  • Dockerfile

  • GitHub copilot

  • jupyter and ipython

  • Most common Python libraries for ML/DL and Hugging Face

  • githubactions

Verify GPU works

The following examples test out the GPU

  • run python utils/quickstart_pytorch.py
  • run python utils/verify_cuda_pytorch.py
  • run nvidia-smi -l 1 it should show a GPU
  • run whisper utils/four-score.m4a --model large and verify GPU is working with nvidia-smi -l 1

Used in Following Projects

Used as the base and customized in the following Duke MLOps and Applied Data Engineering Coursera Labs:

References

About

mlops template

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 87.8%
  • Dockerfile 6.0%
  • Makefile 3.8%
  • Shell 2.4%