pytorch训练时,显存明显上去了,模型和数据都在gpu上,但GPU利用率仍为0%
解决方法:
增加dataloader中的num_workers
train_loader = DataLoader(dataset=train_dataset,
batch_size=512,
shuffle=False, num_workers=6)
pytorch训练时,显存明显上去了,模型和数据都在gpu上,但GPU利用率仍为0%
解决方法:
增加dataloader中的num_workers
train_loader = DataLoader(dataset=train_dataset,
batch_size=512,
shuffle=False, num_workers=6)