首页 > Windows开发 > 详细

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

时间:2016-08-05 00:29:24      阅读:478      评论:0      收藏:0      [点我收藏+]
win10 用cmake 3.5.2 和 vs 2015 update1 编译 GPU版本(cuda 8.0, cudnn v5 for cuda 8.0)
技术分享
 用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里面复制过去,以下几个文件:
技术分享
 
======
技术分享
 ==============
以及根目录有一个设置环境变量的setupenv.cmd技术分享 
还有LICENSE(从原工程拷过去)
README.txt(有些过时了)

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

不过有个问题,在我的电脑上跑mlp_numpy的例子,gpu比cpu慢。。。
技术分享
 技术分享
 








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

原文:http://www.cnblogs.com/guwang/p/5738733.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!