安装mxnet GPU
sudo pip install mxnet-cu80==1.1.0
(1)使用系统python验证，命令行中输入：python
1，cpu：
from mxnet import nd
x = nd.array([1,2,3])
x.context
cpu(0)
2，GPU
from mxnet import nd
a = nd.array([1,2,3], ctx=mx.gpu())
a
[ 1. 2. 3.]
<NDArray 3 @gpu(0)>,
原文：https://www.cnblogs.com/adong7639/p/8954476.html