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

相关文章:

  • 2022-12-23
  • 2021-08-21
  • 2021-06-06
  • 2018-12-09
  • 2022-12-23
  • 2021-11-27
  • 2022-12-23
  • 2022-02-16
猜你喜欢
  • 2021-10-12
  • 2021-06-17
  • 2022-12-23
  • 2022-12-23
  • 2021-08-21
  • 2021-12-18
  • 2022-12-23
相关资源
相似解决方案