torch版本问题:2.0.0不行,1.11.0可以。 上次是DataLoader.pin_memory_device设为True保存为pkl,再次调用设备不一致,保存时pin_memory_device设为False即可