GPU提升多分类问题

使用GPU加速多分类问题，与上一篇文章中前几部分代码相同
在这里插入图片描述

多层线性网络这里，直接使用nn.Module搭建，与之前自定义搭建的三层网络一样，但注意这里用的激活函数是ReLU的改良后的LeakyReLU，能够避免在输入小于0时出现梯度弥散。
在这里插入图片描述
设置GPU设备，并把网络和loss函数计算搬到GPU上

数据也搬到GPU上

在任务管理器上看代码运行前后效果，验证是否使用了GPU

相关阅读:
Xilinx 7系列FPGA的配置流程
【知识点随笔分析 | 第六篇】HTTP/1.1，HTTP/2和HTTP/3的区别
金仓数据库 KingbaseES异构数据库移植指南 (2. 概述）
Qt中的信号和槽详解
mybatis-puls使用修改语句部分字段不生效
Element UI表格将两列数据放在一起显示
用huggingface.Accelerate进行分布式训练
jasypt组件死锁bug案例分享
2022全球智慧城市大会升哲科技：以人为核心推动智慧城市建设
百日刷题计划 ———— DAY1

原文地址：https://blog.csdn.net/qq_52015311/article/details/133772294