Kaldi不支持Cuda10_1版本

請注意時效性。

 

筆者近期下載了Kaldi5.3+Cuda10.1， 系統是Ubuntu16.

編譯Kaldi時提示

Unsupported CUDA_VERSION (CUDA_VERSION=10_1)， please report it to Kaldi mailing list, together with 'nvcc -h' or 'ptxas -h' which lists allowded -gencode values...

 

 

Google了一下原因，可能時5.3版本時還沒有出Cuda，所以在當時沒有列出Cuda10版本。需要修改一下configure文件。

在configure 410行，新增10_*

9_2 | 9_* | 10_*)

在430行，新增如下

 10_*) CUDA_ARCH="-gencode arch=compute_30,code=sm_30 -gencode arch=compute_35,code=sm_35 -gencode arch=compute_50,code=sm_50 -gencode arch=compute_53,code=sm_53 -gencode arch=compute_60,code=sm_60 -gencode arch=compute_61,code=sm_61 -gencode arch=compute_62,code=sm_62 -gencode arch=compute_70,code=sm_70 -gencode arch=compute_72,code=sm_72 -gencode arch=compute_75,code=sm_75" ;;

截圖如下

 

參考git提交記錄

https://github.com/kaldi-asr/kaldi/pull/2747/commits/0279a16e1fdabcda9e04442ecb80af4863ec1495