首页 > 数据库技术 > 详细

CUDA6.5 + GTX960 + Winpython,运行Theano的DBN

时间:2015-07-23 23:36:30      阅读:1141      评论:0      收藏:0      [点我收藏+]

按照前面的帖子安装好之后,在几台机器上运行,运算速度简单记了一下。

 

1. Win7, i7-4790K, 32G (8*4)

纯CPU计算,第一遍预训练每步时间:42.x秒

 

2. Y480笔记本电脑,4G

GPU加速计算,第一遍预训练每步时间:90.x秒

 

[新显卡支持问题的解决]

之后购买了影驰GTX960 4G,运行时提示错误:

nvcc fatal   : Value ‘sm_52‘ is not defined for option ‘gpu-architecture‘

网上搜索发现了答案:

http://rexdouglass.com/training-neural-networks-on-the-gpu-with-commodity-hardware-installation-and-configuration/

只要在.theanorc设置文件中加上如下内容即可:

[nvcc]
flags = -arch=sm_30

原因是CUDA6.5版本太低,显卡太新,据说换成7.0或7.5即可,我嫌麻烦,没换。

也许7.x版本的能提升速度?以后有时间再试试。

 

然后在两台台式机上实验:

 

3. Win7, i5-3450, 8G (8*1)

GPU加速计算,第一遍预训练每步时间:19.x秒

 

4. Win7, i7-4790K, 32G (8*4)

GPU加速计算,第一遍预训练每步时间:15.x秒

 

总结:

1. i7-4790K很强大

2. Y480笔记本电脑GPU很弱

3. i7台式机比i5台式机快,可能是因为内存快,同时i7的用上了双通道

 

CUDA6.5 + GTX960 + Winpython,运行Theano的DBN

原文:http://www.cnblogs.com/theagnal/p/4671955.html

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