GPU Memory Usage佔用而GPU-Util卻為0

仁義禮智信達發表於2020-09-24

新增num_workers=num,需要多設定一些,就可以使用GPU加速

train_loader = DataLoader(dataset=train_data, batch_size=train_bs, shuffle=True, num_workers=8)

 

相關文章