Windows下安装mxnet_cu90

最近无聊发现李沐大神的深度学习资料,于是也想着手学习一下mxnet,本以为只是 在anaconda下简单的

pip install mxnet_cu90,但是发现不能成功调用GPU,弄了大半天也不行,最终找到如下可用办法:

 

1   git clone https://github.com/mli/gluon-tutorials-zh

2  conda create -n gluon anaconda python=3.6

3 activate gluon

4 pip install -i  https://pypi.douban.com/simple/ --pre mxnet-cu90

 

由于我装的是cuda90,所以后缀是cu_90,各位可以根据自己的cuda版本进行更改,还有就是。。。豆瓣是真的快,比清华的,科大的快好多倍。。。可惜今天才发现

 

装好之后可以测试一下是否可以成功调用GPU,但是要提前装好CUDA和cudnn哦

 import mxnet as mx

a = mx.nd.array([1,2,3],ctx=mx.gpu())

a

结果显示:

Windows下安装mxnet_cu90

另外附上官网和:

http://mxnet.incubator.apache.org/

一本很好的深度学习book:

 http://neuralnetworksanddeeplearning.com/index.html

 

Talk is cheap ,show me the code....