07 2022 档案

检测是否torch的CUDA是否可用的代码
摘要：import torch print(torch.cuda.is_available()) # 返回True 接着用下列代码进一步测试 print(torch.zeros(1).cuda()) 阅读全文
posted @ 2022-07-06 17:13 乌蝇哥 阅读(598) 评论(0) 推荐(0) 编辑
torch训练和推理的模板
摘要：def train(epoch): model.train() train_loss = 0 for data, label in train_loader: data, label = data.cuda(), label.cuda() # 将数据放入显卡 optimizer.zero_grad( 阅读全文
posted @ 2022-07-01 20:12 乌蝇哥 阅读(388) 评论(0) 推荐(0) 编辑

点击右上角即可分享
微信分享提示