search

649453932 / Chinese-Text-Classification-Pytorch

中文文本分类，TextCNN，TextRNN，FastText，TextRCNN，BiLSTM_Attention，DPCNN，Transformer，基于pytorch，开箱即用。
MIT License
5.19k stars 1.22k forks source link

GPU训练问题 #80

Open li199603 opened 2 years ago

li199603 commented 2 years ago

首先感谢提供的代码。 使用GPU训练反而慢了，而且慢很多。这是BUG吗？还是只有我是这样情况

MichaelMachine commented 2 years ago

我也出现了这样的问题

reatiny commented 2 years ago

请问发现是什么原因了吗

hhqd commented 2 years ago

检查一下cuda驱动，正常GPU是快很多的 torch.cuda.is_available()

reatiny commented 2 years ago

发现用pytorch1.4 GPU 就无问题了