[CUDA手搓]从零开始用C++ CUDA搭建一个卷积神经网络(LeNet),了解神经网络各个层背后算法原理
文章目录前言一、所需环境二、实现思路2.1.定义了LeNet网络模型结构,并训练了20次2.2以txt格式导出训练结果(模型的各个层权重偏置等参数)2.3(可选)以pth格式导出训练结果,以方便后期调试2.4C++CUDA要做的事三、C++CUDA具体实现3.1新建.cu文件并填好框架3.2C++实现各网络层3.0CUDA编程核心思路3.1卷积层Conv13.2激活函数ReLu13.2池化层Max