2022年10月7日
利用GPU进行模型训练
摘要： GPU训练方式1 网络模型 数据（输入，标注） 损失函数 加上.cuda() GPU训练方式2 # 测试是否有gpu device = torch.device('cuda' if torch.cuda.is_available() else 'cpu') # 如果有多张显卡，可以指定显卡 # de 阅读全文
posted @ 2022-10-07 14:02 watsono 阅读(58) 评论(0) 推荐(0) 编辑