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

全部代码依托于该博客

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

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

代码更改如下:

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

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

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

代码更改如下:



相关推荐
itzixiao几秒前
L1-066 猫是液体(5分)[java][python]
java·开发语言·python·算法
zhoutongsheng4 分钟前
如何解决ORA-01078参数文件错误_pfile与spfile互相创建恢复
jvm·数据库·python
小饕5 分钟前
从 Word2Vec 到多模态:词嵌入技术的演进全景
人工智能·算法·机器学习
上海云盾第一敬业销售5 分钟前
生成式AI催生深度伪造攻击,WAF如何识别“假流量“?
人工智能
ykjhr_3d7 分钟前
数字工具AI智能学伴,助力教育数字化转型
大数据·人工智能·ai·ai人工智能·华锐视点·华锐云空间
Lightning-py7 分钟前
Python 配置日志(Logging)
开发语言·python
2401_824222698 分钟前
HTML怎么标注字数限制提示_HTML实时字数统计占位【详解】
jvm·数据库·python
LIUAWEIO9 分钟前
鸽鸽工具网:免费在线工具大全,打开网页即用
人工智能·安全·ai·json
动恰客流管家12 分钟前
动恰3DV3丨客流统计系统:旺季人手不够淡季闲人太多?客流统计帮你科学优化人力成本
大数据·运维·人工智能·3d
d111111111d12 分钟前
了解Modbus
网络·笔记·stm32·单片机·嵌入式硬件·学习