- Sandbox to create MLops pipeline
- 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