windows 10下安装GPU版MXNet

11 篇文章 0 订阅
9 篇文章 0 订阅
上回说到编译CPU版的MXNet: [url]http://cherishlc.iteye.com/blog/2299864[/url]
最近攒了个带GPU的本本,,终于可以整GPU版的啦~

过程和编译CPU版的差不多,但是却花了3天时间填坑,,具体过程本文不再详述,请先编译CPU版。
与编译CPU版不同之处在于:
[list]
[*]win10下必须使用Cuda8RC(我的是965m的GPU),虽然Cuda7.5有win10版本,实测不兼容!!!结果下Cuda8总是不成功,花了一天
[*]CMake需要指定编译64位平台的代码!!参见:[url]http://cherishlc.iteye.com/blog/2302987[/url]
[*]DLL代码要加到环境变量PATH里,上次说过了,这次居然又栽跟头了。。。
[/list]

具体流程参加CPU版的,,本文只是对编译GPU版本踩过的坑进行记录
[size=large][color=blue]1、Cuda、cuDNN的安装[/color][/size]
Cuda:[url]https://developer.nvidia.com/cuda-toolkit[/url]
一路狂点即可,注意下载Cuda8RC版,,否则可能有兼容问题

cuDNN:[url]https://developer.nvidia.com/cudnn[/url]
要用v3版本的(MXNet官网这么说的,不知道现在是否支持v4、v5)。
记得把cuDNN的DLL所在目录加到PATH里!(和CPU版的openBLAS类似)


[size=large][color=blue]2、OpenCV的编译[/color][/size]
由于换了全新的本本,OpenCV我也重新编译了一下,结果栽了跟头!
[list]
[*]注意把VS的工程里的编译模式debug改为release!!!
[*]DLL所在路径添加到PATH,,上次没有这一步,这次就栽了跟头。。。表现为运行时找不到模块,而且不会提示找不到什么模块。。。
[*]编译GPU版时间老长了,,没必要。。。
[/list]

[size=large][color=blue]3、MXNet工程的生成[/color][/size]
上次说到直接用CMake的CMake-gui工具配置,这次出了问题,,没生成x64平台的,只有win32平台。。。还是编译时发现一堆明明存在的lib却找不到时才发现的!!!
再次安利CMake编译x64平台代码的博文:[url]http://cherishlc.iteye.com/blog/2302987[/url]

即使一切配置正确,,编译还是不通过,提示找不到[b]opencv_dep_cudart.lib[/b],当时真是想重新编译一个CPU版的OpenCV!但是,当我将该lib从MXNet工程的输入中剔除后,编译通过了!!

只是运行时提示找不到库(还没告诉我找不到什么库。。。),后来才想起来OpenCV的DLL路径没添加到PATH

再遇到这种情况可以使用depends工具查看DLL依赖关系:[url]http://www.dependencywalker.com/[/url]
其中,windows相关的组件找不到貌似没啥关系,,因为我添加了OpenCV路径后依然会说windows相关的组件找不到

[size=large][color=blue]4、安装python版的MXNet[/color][/size]
和CPU版完全相同,,不再赘述


[size=large][color=blue]5、运行示例[/color][/size]
安装好了就可以参照PHunter的博文跑一下[b]Neural art[/b]了:[url]http://phunter.farbox.com/post/mxnet-tutorial2[/url]
只是需要注意,由于windows无法运行.sh文件,我们需要手动下载模型

windows下[b]nvidia-smi[/b]程序在:[i]C:\Program Files\NVIDIA Corporation\NVSMI[/i] 下
官方文档:[url]https://developer.nvidia.com/nvidia-system-management-interface[/url]

拿左图同时作为输入和style image的输出居然和他自己不一样。。。
[img]http://dl2.iteye.com/upload/attachment/0117/8652/f09f0c0b-1fa7-3d23-84ca-f5bc481791bf.jpg[/img]
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值