当前位置: 代码迷 >> 综合 >> GPU Memory Usage占用而GPU-Util却为0
  详细解决方案

GPU Memory Usage占用而GPU-Util却为0

热度:0   发布时间:2024-02-22 05:22:49.0

添加num_workers=num,需要多设置一些,就可以使用GPU加速

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

 

  相关解决方案