mxnet的设备管理 MXNet 使用 context 来指定用来存储和计算的设备,例如可以是 CPU 或者 GPU.默认情况下,MXNet 会将数据创建在主内存,然后利用 CPU 来计算.在 MXNet 中,CPU 和 GPU 可分别由 cpu() 和 gpu() 来表示. 需要注意的是,mx.cpu()(或者在括号里填任意整数)表示所有的物理 CPU 和内存.这意味着计算上会尽量使用所有的 CPU 核. 但 mx.gpu() 只代表一块显卡和相应的显卡内存.如果有多块 GPU,我们用 mx.…