ubuntu安正mxnet GPU版本

标签：==   contex   验证   pytho   输入   array   python   context   ubuntu   

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

ubuntu安正mxnet GPU版本

标签：==   contex   验证   pytho   输入   array   python   context   ubuntu   

原文地址：https://www.cnblogs.com/adong7639/p/8954476.html