深度学习

PyTorch,torchvision版本对应关系

PyTorch,torchvision版本对应关系

概述

PyTorch 和 torchvision 之间存在版本依赖关系。选择兼容的版本组合对于项目的稳定性至关重要。

版本对应表

PyTorchtorchvisionPython
2.9.x0.24.x3.10-3.14
2.8.x0.23.x3.9-3.13
2.7.x0.22.x3.9-3.13
2.6.x0.21.x3.8-3.12
2.5.x0.20.x3.8-3.12
2.4.x0.19.x3.8-3.12
2.3.x0.18.x3.8-3.12
2.2.x0.17.x3.8-3.12
2.1.x0.16.x3.8-3.11
2.0.x0.15.x3.8-3.11
1.13.x0.14.x3.7-3.11
1.12.x0.13.x3.7-3.10
1.11.x0.12.x3.7-3.10
1.10.x0.11.x3.6-3.10
1.9.x0.10.x3.6-3.10

安装建议

使用 pip 安装

cpu版本安装

1

pip install torch==2.5.0 torchvision==0.20.0

gpu版本安装 可根据需求更改版本

1

pip3 install torch==2.5.0+cu118 torchvision==0.20.0+cu118  --index-url https://download.pytorch.org/whl/cu118

验证安装

1

python -c "import torch; import torchvision; print(f'PyTorch: {torch.__version__}'); print(f'torchvision: {torchvision.__version__}')"

测试gpu是否可用

1
2

import torch
torch.cuda.is_available()

官方资源

© 2025 - 2026 topqaz