2022年5月27日
30-31 利用GPU训练
摘要： 一、第一种方法 1.找到能够调用.cuda（）的位置 网络模型 数据（输入，标注） 损失函数 2.计算训练的时间 在CPU运行速度 代码： 点击查看代码 import torch import torchvision # 13. 计算时间 import time from torch.utils.t 阅读全文
posted @ 2022-05-27 11:27 Trouvaille_fighting 阅读(249) 评论(0) 推荐(0)