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....

 

 

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值