ubuntu安装mxnet GPU版本

安装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)>,

posted on 2018-04-26 19:01  Maddock  阅读(1302)  评论(0编辑  收藏  举报

导航