【报错】RuntimeError: CUDA error: an illegal memory access was encountered【解决方法】DEVICE = torch.device("cuda:2") 改成 torch.cuda.set_device(2)