Pytorch学习--神经网络--利用GPU训练

全部代码依托于该博客

一、第一种方法:XX.cuda()

模型、数据+标签、损失函数都放到 gpu 上面

代码更改如下:

二、第二种方法:XX.to(device) --更推荐

device = torch.device("cuda" if torch.cuda.is_available() else "cpu")

同样将模型、数据+标签、损失函数都放到 device 上面

代码更改如下:



相关推荐
盐水冰1 分钟前
【烘焙坊项目】后端搭建(14) - 工作台&导出数据报表
java·后端·学习
小鹿软件办公1 分钟前
谷歌目前正在测试原生 Mac 版 Gemini 客户端
人工智能·gemini
Deepoch1 分钟前
Deepoc具身模型开发板:构建机械臂柔性制造的通用“神经中枢”
人工智能·科技·机械臂·具身模型·deepoc
人工智能AI技术3 分钟前
OpenAI超级App合并三端!GPT+Codex一体化开发实战
人工智能
曲辕RPA5 分钟前
深度解析GEO技术及背后的机器人曲辕RPA
python·ai·自动化·rpa
旗讯数字5 分钟前
服装吊牌智能识别+结构化抽取+国标合规审查|旗讯数字解决方案
大数据·人工智能
ZWZhangYu5 分钟前
【Gradio系列】快速入门
人工智能
无代码专家6 分钟前
零代码平台 2026 发展报告:轻流 AI 重塑业务流程管理
人工智能·低代码
ZPC82109 分钟前
【无标题】
人工智能·pytorch·算法·机器人
人工智能AI技术9 分钟前
华为AgentArts公测|企业级AI智能体开发与openJiuwen适配指南
人工智能