如何安装torchvision
时间: 2023-11-24 16:03:05 浏览: 73
Linux系统安装pytorch和torchvision:torch1.8.0+torchvision0.9.0
Torchvision是PyTorch的一个扩展库,提供了许多计算机视觉相关的工具和数据集。安装Torchvision需要先安装PyTorch。
以下是在Windows系统下使用pip安装PyTorch和Torchvision的步骤:
1. 确认已经安装了Python3和pip,可以在命令行中输入以下命令检查:
```
python --version
pip --version
```
2. 安装PyTorch。打开命令行,输入以下命令:
```
pip install torch
```
如果电脑配置较低,建议下载CPU版本的PyTorch,命令为:
```
pip install torch-cpu
```
3. 安装Torchvision。打开命令行,输入以下命令:
```
pip install torchvision
```
安装完成后,可以使用以下命令检查是否安装成功:
```
import torch
import torchvision
print(torch.__version__)
print(torchvision.__version__)
```
输出版本号即为安装成功。
阅读全文