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
结果显示:
另外附上官网和:
http://mxnet.incubator.apache.org/
一本很好的深度学习book:
http://neuralnetworksanddeeplearning.com/index.html
Talk is cheap ,show me the code....