使用gpu加速训练
使用torch.cuda.is_available()检查gpu是否可用
创建model后使用model.to('cuda')将模型放到gpu上
将输入x通过 x.to('cuda')放到gpu上
将标签y通过 y.to('cuda')放到gpu上
其它不变即可从cpu上训练转为gpu上训练
使用gpu加速训练
使用torch.cuda.is_available()检查gpu是否可用
创建model后使用model.to('cuda')将模型放到gpu上
将输入x通过 x.to('cuda')放到gpu上
将标签y通过 y.to('cuda')放到gpu上
其它不变即可从cpu上训练转为gpu上训练