安装mxnet GPU
sudo pip install mxnet-cu80==1.1.0
推荐pip安装mxnet,土豪gpu版本:
pip install mxnet-cu90==1.0.0
豪华至尊gpu+mkl版本
pip install mxnet-cu90mkl==1.0.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)>,