torch==2.1.2
torchvision==0.16.2
torchaudio==2.1.2