高CPU消耗-PyTorch

Question

尽管我看到了一些有关我的问题的问题/答案，但仍无法解决。我正在尝试从GitHub运行基本代码来培训GAN。尽管代码在GPU上运行，但在训练过程中CPU使用率为100％（甚至更高）。

https://github.com/eriklindernoren/PyTorch-GAN/blob/master/implementations/wgan_gp/wgan_gp.py

为了使用我的数据，我添加了以下在PyTorch官方文档中检查的数据加载代码。

def Read_LMDB(root, classes):
    d1 = torchvision.datasets.LSUN(root=root, classes=classes, transform=transforms.Compose(
                [transforms.Resize((opt.img_size, opt.img_size)), transforms.ToTensor(), transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5))]))
    dataloader = torch.utils.data.DataLoader(d1, batch_size=opt.batch_size, shuffle=True)
    return dataloader

def Read_from_folder(folder_name):
    imagefolder = datasets.ImageFolder(folder_name,transform=transforms.Compose(
        [transforms.Resize(opt.img_size), transforms.ToTensor(), transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5))]))
    dataloader = torch.utils.data.DataLoader(imagefolder,batch_size=opt.batch_size,shuffle=True)
    return dataloader

但是，仍然是同样的问题。我使用哪个功能都没有关系，CPU使用率是100％。使用如此高的CPU是不正常的，因为训练是在GPU上进行的（GPU使用没有问题。GPU内存使用，实用性和训练速度都可以）。

有人可以有任何想法吗？谢谢！

最好。