ASRT_SpeechRecognition GPU训练速度

GPU训练速度

Open ASlepnir opened this issue 4 years ago • 3 comments

想知道作者训练时是用的什么规格的GPU？这边想训练自己的模型，但是200h+有些太长，所以在考虑增加多块2080Ti比较好还是提高到Titan比较好呢？

Mar 29 '20 11:03 ASlepnir

还是直接多块2080Ti吧，速度提升能更明显一些

Mar 29 '20 14:03 nl8590687

请问我有4块2080ti，该怎么配置多卡训练，我试着改os.environ["CUDA_VISIBLE_DEVICES"] = "0，1,2,3"，但是只有第一个gpu在使用，batch>16就会显存溢出

Apr 23 '20 06:04 thunder95

请问我有4块2080ti，该怎么配置多卡训练，我试着改os.environ["CUDA_VISIBLE_DEVICES"] = "0，1,2,3"，但是只有第一个gpu在使用，batch>16就会显存溢出

改用github上面带"_p"的那些模块，在train_mspeech里面导入后去训练就行 @thunder95

Apr 25 '20 06:04 nl8590687