本程序基于PyTorch,需要从官网下载指定版本的PyTorch(2.7,CUDA).