tmbdev / clstm

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

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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

moucmou opened this issue · comments

commented

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 !!