版本配套

TorchNPU需与特定版本的PyTorch、CANN、Python及固件与驱动配合使用。各版本间的匹配关系由昇腾社区统一维护,建议优先使用推荐版本组合以确保最佳兼容性和性能。固件与驱动的版本配套与所使用的昇腾硬件及CANN版本相关,具体选择请参考 固件与驱动安装页面

推荐版本组合

组件 推荐版本
TorchNPU 2.12.0(安装包版本)
PyTorch 2.12.0
CANN 9.1.0
Python 3.10 / 3.11 / 3.12 / 3.13 / 3.14

TorchNPU 版本配套表

TorchNPU发布版本的分支名称采用 {PyTorch版本}-{TorchNPU版本} 命名规则,前者为TorchNPU匹配的PyTorch版本,以下为当前活跃版本的匹配关系:

TorchNPU安装包版本 GitCode分支 PyTorch版本 CANN版本
2.12.0 v2.12.0-26.1.0 2.12.0 CANN 9.1.0
2.11.0 v2.11.0-26.1.0 2.11.0 CANN 9.1.0
2.10.0.post4 v2.10.0-26.1.0 2.10.0 CANN 9.1.0
2.9.0.post6 v2.9.0-26.1.0 2.9.0 CANN 9.1.0
2.7.1.post8 v2.7.1-26.1.0 2.7.1 CANN 9.1.0
2.9.0.post2 v2.9.0-26.0.0 2.9.0 CANN 9.0.0
2.8.0.post4 v2.8.0-26.0.0 2.8.0 CANN 9.0.0
2.7.1.post4 v2.7.1-26.0.0 2.7.1 CANN 9.0.0
2.9.0 v2.9.0-7.3.0 2.9.0 CANN 8.5.0
2.8.0.post2 v2.8.0-7.3.0 2.8.0 CANN 8.5.0
2.7.1.post2 v2.7.1-7.3.0 2.7.1 CANN 8.5.0
2.6.0.post5 v2.6.0-7.3.0 2.6.0 CANN 8.5.0
2.8.0 v2.8.0-7.2.0 2.8.0 CANN 8.3.RC1
2.7.1 v2.7.1-7.2.0 2.7.1 CANN 8.3.RC1
2.6.0.post3 v2.6.0-7.2.0 2.6.0 CANN 8.3.RC1
2.1.0.post17 v2.1.0-7.2.0 2.1.0 CANN 8.3.RC1
2.6.0 v2.6.0-7.1.0 2.6.0 CANN 8.2.RC1
2.5.1.post1 v2.5.1-7.1.0 2.5.1 CANN 8.2.RC1
2.1.0.post13 v2.1.0-7.1.0 2.1.0 CANN 8.2.RC1
2.5.1 v2.5.1-7.0.0 2.5.1 CANN 8.1.RC1
2.4.0.post4 v2.4.0-7.0.0 2.4.0 CANN 8.1.RC1
2.3.1.post6 v2.3.1-7.0.0 2.3.1 CANN 8.1.RC1
2.1.0.post12 v2.1.0-7.0.0 2.1.0 CANN 8.1.RC1
2.4.0.post2 v2.4.0-6.0.0 2.4.0 CANN 8.0.0
2.3.1.post4 v2.3.1-6.0.0 2.3.1 CANN 8.0.0
2.1.0.post10 v2.1.0-6.0.0 2.1.0 CANN 8.0.0
点击展开历史版本(含 EOL)
TorchNPU安装包版本 GitCode分支 PyTorch版本 CANN版本
2.4.0 v2.4.0-6.0.rc3 2.4.0 CANN 8.0.RC3
2.3.1.post2 v2.3.1-6.0.rc3 2.3.1 CANN 8.0.RC3
2.1.0.post8 v2.1.0-6.0.rc3 2.1.0 CANN 8.0.RC3
2.3.1 v2.3.1-6.0.rc2 2.3.1 CANN 8.0.RC2
2.2.0.post2 v2.2.0-6.0.rc2 2.2.0 CANN 8.0.RC2
2.1.0.post6 v2.1.0-6.0.rc2 2.1.0 CANN 8.0.RC2
1.11.0.post14 v1.11.0-6.0.rc2 1.11.0 CANN 8.0.RC2
2.2.0 v2.2.0-6.0.rc1 2.2.0 CANN 8.0.RC1
2.1.0.post4 v2.1.0-6.0.rc1 2.1.0 CANN 8.0.RC1
1.11.0.post11 v1.11.0-6.0.rc1 1.11.0 CANN 8.0.RC1
2.1.0 v2.1.0-5.0.0 2.1.0 CANN 7.0.0
2.0.1.post1 v2.0.1-5.0.0 2.0.1 CANN 7.0.0
1.11.0.post8 v1.11.0-5.0.0 1.11.0 CANN 7.0.0
2.1.0.rc1 v2.1.0-5.0.rc3 2.1.0 CANN 7.0.RC1
2.0.1 v2.0.1-5.0.rc3 2.0.1 CANN 7.0.RC1
1.11.0.post4 v1.11.0-5.0.rc3 1.11.0 CANN 7.0.RC1
1.11.0.post3 v1.11.0-5.0.rc2.2 1.11.0 CANN 6.3.RC3.1
1.11.0.post2 v1.11.0-5.0.rc2.1 1.11.0 CANN 6.3.RC3
2.0.1.rc1 v2.0.1-5.0.rc2 2.0.1 CANN 6.3.RC2
1.11.0.post1 v1.11.0-5.0.rc2 1.11.0 CANN 6.3.RC2
1.8.1.post2 v1.8.1-5.0.rc2 1.8.1 CANN 6.3.RC2
1.11.0 v1.11.0-5.0.rc1 1.11.0 CANN 6.3.RC1
1.8.1.post1 v1.8.1-5.0.rc1 1.8.1 CANN 6.3.RC1
1.5.0.post8 v1.5.0-3.0.0 1.5.0 CANN 6.0.1
1.8.1 v1.8.1-3.0.0 1.8.1 CANN 6.0.1
1.11.0.rc2(beta) v1.11.0-3.0.0 1.11.0 CANN 6.0.1
1.5.0.post7 v1.5.0-3.0.rc3 1.5.0 CANN 6.0.RC1
1.8.1.rc3 v1.8.1-3.0.rc3 1.8.1 CANN 6.0.RC1
1.11.0.rc1(beta) v1.11.0-3.0.rc3 1.11.0 CANN 6.0.RC1
1.5.0.post6 v1.5.0-3.0.rc2 1.5.0 CANN 5.1.RC2
1.8.1.rc2 v1.8.1-3.0.rc2 1.8.1 CANN 5.1.RC2
1.5.0.post5 v1.5.0-3.0.rc1 1.5.0 CANN 5.1.RC1
1.8.1.rc1 v1.8.1-3.0.rc1 1.8.1 CANN 5.1.RC1
1.5.0.post4 2.0.4.tr5 1.5.0 CANN 5.0.4
1.5.0.post3 2.0.3.tr5 1.8.1 CANN 5.0.3
1.5.0.post2 2.0.2.tr5 1.5.0 CANN 5.0.2

PyTorch与Python版本配套表

PyTorch版本 Python版本
PyTorch 2.13.0 Python3.10, Python3.11, Python 3.12, Python 3.13, Python 3.14
PyTorch 2.12.0 Python3.10, Python3.11, Python 3.12, Python 3.13, Python 3.14
PyTorch 2.11.0 Python3.10, Python3.11, Python 3.12, Python 3.13, Python 3.14
PyTorch 2.10.0 Python3.10, Python3.11, Python 3.12, Python 3.13
PyTorch 2.9.0 Python3.10, Python3.11, Python 3.12, Python 3.13
PyTorch 2.8.0 Python3.9, Python3.10, Python 3.11, Python 3.12, Python 3.13
PyTorch 2.7.1 Python3.9, Python3.10, Python 3.11, Python 3.12, Python 3.13
PyTorch 2.6.0 Python3.9, Python3.10, Python 3.11
PyTorch 2.5.1 Python3.9, Python3.10, Python 3.11
PyTorch 2.4.0 Python3.8, Python3.9, Python3.10, Python 3.11
PyTorch 2.3.1 Python3.8, Python3.9, Python3.10, Python 3.11
PyTorch 2.2.0 Python3.8, Python3.9, Python3.10
PyTorch 2.1.0 Python3.8, Python3.9, Python3.10, Python 3.11
PyTorch 1.11.0 Python3.7(>=3.7.5), Python3.8, Python3.9, Python3.10

硬件支持

TorchNPU支持如下昇腾产品系列:

产品系列
昇腾 950DT 系列产品
昇腾 A3 系列产品
昇腾 A2 系列产品
昇腾 910 系列产品
昇腾 310P 系列产品
昇腾 310B 系列产品
昇腾310系列产品