torch版本问题,最好不要换torch,因为cuda和torch等一系列的版本是绑定的,一旦一个换了,容易连锁出问题。 建议直接换函数 torch.swapaxes 变 torch.transpose