下载
中文
注册

(可选)安装torchvision

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