Pytorch | Pytorch框架中模型和数据的gpu和cpu模式:
model.to(device), model.cuda(), model.cpu(), DataParallel
转载自:https://blog.csdn.net/iLOVEJohnny/article/details/106021547