/mlops_sandbox

sandbox to expereiment with mlops

Primary LanguageJupyter NotebookApache License 2.0Apache-2.0

Introduction

  • Sandbox to create MLops pipeline

Reference

Env Setup

  • PyTorch
  • PyTorch lightning
  • wandb
  • matplotlib
conda install pytorch torchvision torchaudio cudatoolkit=11.6 -c pytorch -c conda-forge
pip install pytorch-lightning
pip install wandb
pip install matplotlib
pip install jupyterlab

# login using key from wandb
wandb login