为什么用tensorflow-gpu建立lstm时没有凸显加速效果,甚至更慢

模型设置:
 
model = Sequential() model.add(LSTM(10,batch_input_shape=(X_train.shape[1],1,X_train.shape[2]),stateful=True, return_sequences=True,activation='linear')) model.add(LSTM(10,activation='linear')) model.add(Dense(1)) start = time.time() model.compile(loss="mse", optimizer="Adam") print("Compilation Time : ", time.time() - start) model.fit(X_train, y_train, epochs=10, batch_size=1,verbose=0, shuffle=False)   提示: 2018-04-07 02:03:02.262652: I T:\src\github\tensorflow\tensorflow\core\common_runtime\gpu\gpu_device.cc:1344] Found device 0 with properties: name: GeForce GTX 1060 major: 6 minor: 1 memoryClockRate(GHz): 1.6705 pciBusID: 0000:01:00.0 totalMemory: 6.00GiB freeMemory: 4.96GiB 2018-04-07 02:03:02.263140: I T:\src\github\tensorflow\tensorflow\core\common_runtime\gpu\gpu_device.cc:1423] Adding visible gpu devices: 0 2018-04-07 02:03:03.242965: I T:\src\github\tensorflow\tensorflow\core\common_runtime\gpu\gpu_device.cc:911] Device interconnect StreamExecutor with strength 1 edge matrix: 2018-04-07 02:03:03.243250: I T:\src\github\tensorflow\tensorflow\core\common_runtime\gpu\gpu_device.cc:917]      0 2018-04-07 02:03:03.243434: I T:\src\github\tensorflow\tensorflow\core\common_runtime\gpu\gpu_device.cc:930] 0:   N 2018-04-07 02:03:03.243786: I T:\src\github\tensorflow\tensorflow\core\common_runtime\gpu\gpu_device.cc:1041] Created TensorFlow device (/job:localhost/replica:0/task:0/device:GPU:0 with 4734 MB memory) -> physical GPU (device: 0, name: GeForce GTX 1060, pci bus id: 0000:01:00.0, compute capability: 6.1)   其中,我看到这个提示跟网上不太一样,虽然能出结果,但我看到网上很多的是 2018-04-07 02:03:03.243434: I T:\src\github\tensorflow\tensorflow\core\common_runtime\gpu\gpu_device.cc:930] 0:   Y 而且预测的结果也很不稳定,一时是全是空值NAN,一时所有预测的都是一样的,有时甚至预测到很大很大的数值,求老师解答,是我代码设置模型错了,还是配置tensorflow+cuda9.0+cudnn+keras的时候少了什么,以至于有这样的问题, 我的测试集容量是6400+,有一个输入变量就是它的滞后一期,然而上面的模型在gpu加速下用了16分钟左右,之前没有tensorflow-gpu,只是tensorflow的两三分钟就可以了,就很郁闷

Robin_TY - PhD@CSU,研究方向:计算机视觉,模式识别

赞同来自: fish

检查下cudnn, cuda和tensorflow的版本是否相互兼容,一般来说用了GPU肯定是快的。 还有就是检查下tensorflow是否安装的是GPU版本的。 也可以参考这里的回答:http://wenda.chinahadoop.cn/question/10056  

小象老师

赞同来自:

请问是哪个课的问题?

要回复问题请先登录注册