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