torch
nvidia-ml-py
