GPU提升多分类问题

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

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

数据也搬到GPU上

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

相关阅读:
力扣日记11.7-【二叉树篇】二叉树的层序遍历
解决CentOS 7 无法连接外网（新手友好版）
发送邮件配置
【已解决】ModuleNotFoundError: No module named ‘dgl‘
[Android Frameworks] native Binder and IMemory
海康机器人工业相机IP设置方式
C++ Reference: Standard C++ Library reference: C Library: cwchar: wcsncmp
【MySQL】实时监控和分析MySQL服务器性能指标的命令
【基础篇】第1章 Elasticsearch 引言
LeetCode 算法：最大子数组和c++

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