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 usedf()# change the default context to the first GPUwith mx.Context(gpu_device): ???f()a = mx.nd.ones((100, 100), gpu_device)aa = mx.nd.ones((100,100), mx.cpu())b = mx.nd.ones((100,100), gpu_device)c = mx.nd.ones((100,100), gpu_device)a.copyto(c) ?# copy from CPU to GPUd = b + ce = b.as_in_context(c.context) + c ?# same to above{‘d‘:d, ‘e‘:e}x = mx.nd.array([[1, 2, 3], [4, 5, 6]])type(x)x.shapey = x + mx.nd.ones(x.shape)*3print(y.asnumpy())z = y.as_in_context(mx.gpu(0))print(z)mxnet-gpu
原文地址：http://blog.51cto.com/13959448/2317243