torch
numpy
nvidia-ml-py3
