Chinese-Text-Classification-Pytorch icon indicating copy to clipboard operation
Chinese-Text-Classification-Pytorch copied to clipboard

Published 20 hours ago • verified publisher icon 649453932

GPU训练问题

Open li199603 opened this issue 3 years ago • 4 comments

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

li199603 avatar Sep 15 '21 08:09 li199603

我也出现了这样的问题

MichaelMachine avatar Nov 18 '21 10:11 MichaelMachine

请问发现是什么原因了吗

reatiny avatar Jun 19 '22 01:06 reatiny

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

hhqd avatar Jun 24 '22 08:06 hhqd

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

reatiny avatar Jun 28 '22 04:06 reatiny