名称错误：名称“torchvision”未定义

Question

我对编码还比较陌生。我正在尝试将 pytorch 用于 cnn 。我已经安装了所有库，版本已在 google colab 中更新，但仍存在 torchvision 未定义的问题

我已经尝试了所有需要做的事情，我只是想使用 torchvision 创建图像数据集。

创建数据集对象

数据集= torchvision.datasets.ImageFolder（image_folder_path，transform=transform）

创建用于批处理的数据加载器

data_loader = DataLoader(dataset, batch_size=32, shuffle=True) # 根据需要调整批量大小

NameError Traceback（最近一次调用最后）

在 () 1 # 创建数据集对象 ----> 2 数据集 = torchvision.datasets.ImageFolder(image_folder_path, 变换=变换) 3 4 # 创建用于批处理的数据加载器 5 data_loader = DataLoader(dataset, batch_size=32, shuffle=True) # 根据需要调整批量大小

名称错误：名称“torchvision”未定义

Answer 1

您必须安装

torchvision

软件包。之后，通过运行

import torchvision

来验证您的安装是否有效。