GPU训练问题
li199603 opened this issue · comments
li199603 commented
首先感谢提供的代码。
使用GPU训练反而慢了,而且慢很多。这是BUG吗?还是只有我是这样情况
keep_original_intention commented
我也出现了这样的问题
wangchenrui commented
请问发现是什么原因了吗
hhqd commented
检查一下cuda驱动,正常GPU是快很多的
torch.cuda.is_available()
wangchenrui commented
发现用pytorch1.4 GPU 就无问题了