win10 用cmake 3.5.2 和 vs 2015 update1 编译 GPU版本(cuda 8.0, cudnn v5 for cuda 8.0)

win10 用cmake 3.5.2 和 vs 2015 update1 编译 GPU版本(cuda 8.0, cudnn v5 for cuda 8.0)
708413-20160804225908606-1178939897.png
 用vs 2015打开
编译Release和Debug版本
看网上那个例子里面
工程里面有是三个文件夹
include(包含mxnet,dmlc,mshadow的include目录)

lib(包含libmxnet.dll, libmxnet.lib,把用vs编译好的放过去)

python(包含一个mxnet,setup.py, 以及build,但是build里面又包含了lib/mxnet 这个跟python里的mxnet库一样,不知道是不是多余的,哦, 这个可能是安装之后才有的)

官方的release还加了一个:3rdparty
(master上面就有一个3rdparty, 但是比如cudnn就要自己更新)
cudart也需要自己从安装的cuda/v8.0/bin里面复制过去,以下几个文件:
708413-20160804225909168-1288617698.png
 
======
708413-20160804225909778-1096829813.png
 ==============
以及根目录有一个设置环境变量的setupenv.cmd  
还有LICENSE(从原工程拷过去)
README.txt(有些过时了)

执行一下环境变量设置的
最后在python目录下执行python setup.py install就可以了

不过有个问题,在我的电脑上跑mlp_numpy的例子,gpu比cpu慢。。。
708413-20160804225911200-1409978525.png
  708413-20160804225911778-742606732.png
 








转载于:https://www.cnblogs.com/guwang/p/5738733.html

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值