649453932 / Chinese-Text-Classification-Pytorch

中文文本分类,TextCNN,TextRNN,FastText,TextRCNN,BiLSTM_Attention,DPCNN,Transformer,基于pytorch,开箱即用。

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

GPU训练问题

li199603 opened this issue · comments

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

我也出现了这样的问题

请问发现是什么原因了吗

commented

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

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