- 博客(3)
- 收藏
- 关注
RSS订阅原创 【动手学深度学习3】实战pytorch中的utils.data.TensorDataset和utils.data.DataLoader工具类
2、DataLoader类：是torch中util.data包下面的一个工具类，用于将TensorDataset整合的数据集（比较大，比如1000个特征和1000个标签）划分为小批量数据级，比如批量大小为10，就会划分为100个。1、TensorDataset类：是torch中util.data包下面的一个工具类，用于把特征和标签整合一个数据集，便于将整个数据集拆分小批量数据集，用于后续小批量梯度下降训练。导入torch和torch.utils库，然后验证输入为一个张量的情况。
2023-11-16 20:30:44
208
1
原创 【动手学深度学习1】Win10 安装CUDA+cuDNN
由于小编的电脑显卡驱动版本为11.7.101，同时为节约电脑硬盘空间，选择下载安装CUDA Toolkit 11.7.0版本。然后选择：帮助--》系统信息--》组件，即可看到显卡驱动版本信息，小编电脑的版本号为11.7.101，表面显卡驱动能支持的cuda最大版本，为11.7。cuDNN是英伟达为深度神经网络打造的一个GPU加速库，使得模型训练推理时速度更快，不是必须的，类似在自行车上装一个电动马达，使得骑得更快。按百度百科定义，CUDA是英伟达推出的一个通用并行计算架构，是为提高GPU计算效率的。
2023-11-05 15:11:27
165
1
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人