折腾了一天半终于装好了win10下的TensorFlow-GPU版在这里做个记录。
按提示安装,先安装cuda_8再安装补丁;
把解压文件放置到CUDA的相关文件夹里:(懒得打字了)
安装过程参照MatConvNet官网给出的步骤:
我先安装了CPU版,然后安装了GPU版:
如果一切顺利(忽略同名警告)编译完后就可以使用cpu版了。
1. 如果你的电脑了里面只有一个CUDNN版本并且與matlab mat版本相匹配,直接用如下代码进行编译:
由于python版本需要使用torch决定先使用matlab mat蝂本试一下,然后就踩进了一些坑以下做一下记录备忘(
MDNet自带一个MatConvnet,需要手动编译自带的MakeFile文件不能实现直接编译,做了一些修改:
在编译的时候尝试使用过cudnn6但是发现似乎不支持,官网上也只说支持cudnn2和4如果之后有需求,再进行更新错误如下:
由于對matlab mat路径不熟悉,使用时将其目录改为了绝对路径
在不进行训练的帧上MDNet在本机运行速度大概是 0.5s/帧 ,其中正向传播每张圖大概用时四分之一秒(256张sample)抽取和保存sample约四份之一秒
若该帧对长期或短期记忆进行训练,总用时大概在1.3s/帧
个人认为虽然追踪效果很好但是时间上消耗还是很厉害。不知道如果使鼡cudnn会不会有效提升效率
版权声明:文章内容来源于网络,版权归原作者所有,如有侵权请点击这里与我们联系,我们将及时删除。