Theano很少使用GPU

时间:2016-03-08 21:19:55

标签: neural-network gpu theano

刚刚开始使用Theano和我的GTX 960,但是当训练网络(特别是循环回归)和GPU使用时间不到一秒钟时,可能需要两到三次才能完成。有没有办法强制它在整个时间内实际使用GPU?

对于没有太多数据似乎需要花费很长时间。

1 个答案:

答案 0 :(得分:1)

你的代码中实际发生的事情是,大部分时间花费在GPU上和从GPU上取下数据,但任何GPU监视器都会显示GPU在此期间“未被使用”,即使它是什么正在做的是加载数据。除了聪明地了解如何以及何时将数据加载到GPU上之外,您无法避免这样做。你描述的尖峰是完全正常的 - 那就是GPU真正进行计算的时候。你无能为力!