(可选)安装torchvision
torchvision为PyTorch的扩展库,主要用于计算机视觉任务。它包括许多常用的模型、数据集,使得处理图像数据变得更加简单高效。
安装PyTorch框架和torch_npu插件后,需安装对应框架版本的torchvision,请参见表1 torchvision版本配套。
执行如下命令安装torchvision,以0.16.0版本为例:
pip3 install torchvision==0.16.0
父主题: 安装步骤