2021年8月22日
pytorch中如何将CPU上运行的数据模型转到GPU上运行（mnist举例）
摘要： 首先贴一份在cpu上运行的代码 1 import torch 2 from torchvision import transforms 3 from torchvision import datasets 4 from torch.utils.data import DataLoader 5 imp 阅读全文
posted @ 2021-08-22 01:07 落悠 阅读(2106) 评论(3) 推荐(0) 编辑