在Pytorch中将张量移动到GPU（使用图像文件夹时）

Question

在使用ImageFolder加载张量后，我试图将张量移至GPU。下面是相关代码：

    train_transform = transforms.Compose([
        transforms.RandomRotation(10),      # rotate +/- 10 degrees
        transforms.RandomHorizontalFlip(),  # reverse 50% of images
        transforms.Resize(224),             # resize shortest side to 224 pixels
        transforms.CenterCrop(224),         # crop longest side to 224 pixels at center
        transforms.ToTensor(),
        transforms.Normalize([0.485, 0.456, 0.406],
                             [0.229, 0.224, 0.225])
    ])

    train_data = datasets.ImageFolder(os.path.join(root, 'train'), transform=train_transform)

这将创建张量的元组，但是当我尝试遍历train_data并使用.cuda（）将张量移动到GPU时，由于元组是不可变的，因此我会报错。

有什么想法可以帮助您吗？谢谢！！