MXNet动手学深度学习笔记：GPU加速计算

1#coding:utf-8
2'''
3NDArray在GPU上计算
4'''
5
6from mxnet import nd
7from mxnet.gluon import nn
8import mxnet as mx
9
10a = nd.array([1,2,3],ctx=mx.gpu())
11b = nd.zeros((3,2),ctx=mx.gpu())
12x = nd.array([1,2,3])
13
14y = x.copyto(mx.gpu())
15
16z = x.as_in_context(mx.gpu())
17
18print('a = ',a)
19print('b = ',b)
20print('x = ',x)
21print('y = ',y)
22print('z = ',z)
23
24# Gluon的GPU计算
25
26net = nn.Sequential()
27net.add(nn.Dense(1))
28net.initialize(ctx = mx.gpu())
29
30data = nd.random.uniform(shape=(3,2),ctx=mx.gpu())
31print(net(data))
32
33# 模型的参数也储存在GPU上
34print(net[0].weight.data())