torch==2.2.1
torchvision==0.17.1
torchaudio==2.2.1

numpy
cython
wheel
ninja