MXNET框架基础3-GPU计算
1、GPU计算,统计时间
2、CPU计算,统计时间
小数据 使用 CPU计算的速度更快。
接下来我们来试一试 大数据,大矩阵的乘法。
3、10000*10000矩阵相乘 cpu计算 用时 33秒
4、设备为GPU 用时 4秒 明显加速
5、指定数据对象所在设备
两种方法:
5.1 创建时
5.2 复制
函数copyto,和 as_in_context
区别在于copyto总是新开一块内存或显存。比如都是在cpu上执行as_in_context,则不会新开内存。
6 计算时一定注意设备一致
提问:
1、大数据集用什么设备,小数据集用什么设备
大数据集,使用 GPU
2、如何在GPU读写模型参数