MXNET框架基础3GPU计算

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读写模型参数

原文地址:https://www.cnblogs.com/lx63blog/p/15729448.html