经典的28*28手写数据集
电脑配置是3070tilaptop和12代i7
在nivdia-msi中查看,同样是个别时候才有高占用率
batch_size设置为64
DataLoader中num_works设置为8,pin_memory设置为truel
神经网络有三个隐层,神经元依次是300 200 100

使用mnist数据搭建的神经网络训练时GPU占用率剧烈波动
- 写回答
- 好问题 0 提建议
- 关注问题
- 邀请回答
-
1条回答 默认 最新
- threenewbee 2023-04-08 12:59关注
mnist数据集太小了,所以gpu用不满,这是很正常的
本回答被题主选为最佳回答 , 对您是否有帮助呢?解决 1无用