在Google Colab用PyTorch构建CNN时遇torchvision未定义错误

• 核心原因是代码中使用了torchvision模块但未提前导入，同时DataLoader也需要从PyTorch的工具模块导入。按以下步骤修复：

1. 添加必要的导入语句
   在代码最开头加入：

   import torchvision
   from torch.utils.data import DataLoader
   

   如果你用到了图像变换（比如代码里的transform），还需要导入torchvision.transforms：

   import torchvision.transforms as transforms
   

2. 验证torchvision安装状态（可选）
   如果导入时仍提示找不到模块，在Colab的代码单元格中执行安装/更新命令：

   !pip install torchvision --upgrade
   

3. 完整可运行的代码片段参考

   import torch
   import torchvision
   import torchvision.transforms as transforms
   from torch.utils.data import DataLoader
   
   # 定义图像预处理变换
   transform = transforms.Compose([
       transforms.Resize((224, 224)),
       transforms.ToTensor(),
       transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225])
   ])
   
   # 创建数据集对象
   dataset = torchvision.datasets.ImageFolder(image_folder_path, transform=transform)
   # 创建批处理数据加载器
   data_loader = DataLoader(dataset, batch_size=32, shuffle=True)
   

内容的提问来源于stack exchange，提问作者Sonu 27