search

srvk / eesen

The official repository of the Eesen project
http://arxiv.org/abs/1507.08240
Apache License 2.0
822 stars 342 forks source link

Add support for CUDA toolkits 7_*, 8_* together with new CUDA compute capabilities #85

Closed standy66 closed 8 years ago

standy66 commented 8 years ago

Hello!

I had troubles running tedlium example script on my GTX 1070. I am using CUDA toolkit 8.0, which is the only supported CUDA toolkit for 1070. After some research, I found out that eesen was compiled with old cuda compute capabilities (namely, arch and code parameters in CUDA_ARCH) which are presumably not supported on my 1070. I looked the way how these parameters are changed in Kaldi in order to support newer CUDA toolkits and tried to reproduce that way in Eesen. After my modifications, tedlium example started working properly.

So I decided that it will be a good idea to merge these changes upstream.

fmetze commented 8 years ago

Great, thanks - this looks like a simple fix. We'll have a few other merges coming up, I think Yajie will take care of them together. Thanks again for sending this!

yajiemiao commented 8 years ago

Thanks! It has been merged.