ubuntu安装mxnet GPU版本

时间:2022-01-23 02:58:24

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