torch
transformers
diffusers
nvidia-smi 
accelerate