torch==2.7.1
torch-npu==2.7.1.post2
torchvision==0.22.1
torchaudio==2.7.1