torch==1.7.1
nvidia-ml-py
