问题

使用model.cuda(),作用是对于模型还是数据，cuda()函数都能实现从CPU到GPU的内存迁移

model.cuda()

报错
Exception has occurred: RuntimeError cuDNN error: CUDNN_STATUS_NOT_INITIALIZED

解决

使用model.to(device)替代，以下0表示GPU的编号，可以在终端下用nvidia-smi查看

device = torch.cuda.set_device(0)#0表示GPU的代码
model = model.to(device)