MxNet学习笔记(2):GPU支持以及其他

GPU的使用

• 在MxNet中，可以通过gpu_device=mx.gpu()来创建GPU的context
• 创建矩阵的时候，可以通过a = mx.nd.ones((100,100))来创建
• 需要执行一个函数的时候，可以通过以下方式在GPU上执行：

gpu_device=mx.gpu() # Change this to mx.cpu() in absence of GPUs.


def f():
    a = mx.nd.ones((100,100))
    b = mx.nd.ones((100,100))
    c = a + b
    print(c)
# in default mx.cpu() is used
f()
# change the default context to the first GPU
with mx.Context(gpu_device):
    f()

• 目前MxNet支持相同平台下的变量运算，如果一个变量在GPU一个变量在CPU,则需要通过copyto之类的方式来统一