Added CUDA 10.1 support from snakers4 patch script

SeanNaren / warp-ctc

Pytorch Bindings for warp-ctc

Apache License 2.0

756 stars 271 forks source link

Added CUDA 10.1 support from snakers4 patch script #159

Closed SeanNaren closed 4 years ago

SeanNaren commented 4 years ago

Replaces #158 due to bugs within the baidu warp-ctc code that I haven't investigated yet. Will look at this simple change and ensure this works via deepspeech.pytorch (thanks @snakers4 and @tq09mx5)

Fixes #157

SeanNaren commented 4 years ago

Works so far, going to merge

s-omranpour commented 4 years ago

Thanks for following up. :)