怎么安装torch1.9.1对应的torchvision的版本
时间: 2023-11-24 12:41:30 浏览: 79
torch-1.5~1.10-cp36-cp36m-linux_aarch64.whl及其torchvision对应版本
5星 · 资源好评率100%
您可以按照以下步骤安装torch1.9.1对应的torchvision的版本:
1. 确认您已经安装了torch1.9.1,可以使用以下命令检查:
```
python -c "import torch; print(torch.__version__)"
```
2. 安装与torch1.9.1对应的torchvision版本,您可以使用以下命令:
```
pip install torchvision==0.10.1+cu111 -f https://download.pytorch.org/whl/cu111/torch_stable.html
```
这将安装与CUDA 11.1兼容的torchvision 0.10.1版本。如果您使用的是其他版本的CUDA,请在上述命令中将“cu111”替换为相应的CUDA版本。
如果您使用的是CPU版本的torch,请使用以下命令安装torchvision:
```
pip install torchvision==0.10.1
```
3. 确认您已经安装成功,可以使用以下命令检查:
```
python -c "import torchvision; print(torchvision.__version__)"
```
如果输出版本号为0.10.1,则表示已经成功安装了torchvision。
阅读全文