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