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