torch
transformers
gradio
nvidia-smi