解决方法1:cuda 版本和pytorch版本不匹配,换版本 解决办法2: work_num 或者 batch_size设置的小一点 解决方法3:torch.backends.cudnn.enabled=False,运行速度会慢