torch==2.2.1+cu121
numpy
matplotlib
