I want to use gpu to speed up training model! How do I do, I follow the previous issu to do but failed

tmbdev / clstm

A small C++ implementation of LSTM networks, focused on OCR.

Apache License 2.0

821 stars 224 forks source link

I want to use gpu to speed up training model! How do I do, I follow the previous issu to do but failed #142

Closed moucmou closed 6 years ago

moucmou commented 6 years ago

I have used scons & scons install compiled successfully, but unfortunately it did not use gpu,then i use scons -j 4 gpu=1 clstmocrtrain clstmocr,Prompt me the following error！！ wrong my gpu driver driver and my cuda nvcc: NVIDIA (R) Cuda compiler driver Copyright (c) 2005-2017 NVIDIA Corporation Built on Fri_Sep__1_21:08:03_CDT_2017 Cuda compilation tools, release 9.0, V9.0.176

Can anyone help me? Sincere thanks ！！