文档HOT
下载
中文
注册

版本配套关系

Ascend Extension for PyTorch版本号采用{PyTorch版本}-{昇腾版本}命名规则，前者为Ascend Extension for PyTorch匹配的PyTorch版本，后者用于匹配CANN版本，详细匹配如下表：
表1 PyTorch软件版本配套关系

CANN版本

PyTorch版本

Ascend Extension for PyTorch版本

Ascend Extension for PyTorch代码分支名称

Ascend Extension for PyTorch安装包版本

Python版本

商发版：8.0.0

社区版：8.0.0.beta1

商发版：8.0.RC3.20

2.1.0

6.0.0

v2.1.0-6.0.0

2.1.0.post10

Python3.8.x、Python3.9.x、Python3.10.x、Python3.11.x

2.3.1

v2.3.1-6.0.0

2.3.1.post4

2.4.0

v2.4.0-6.0.0

2.4.0.post2

商发版：8.0.RC3

社区版：8.0.RC3.beta1

2.1.0

6.0.rc3

v2.1.0-6.0.rc3

2.1.0.post8

Python3.8.x、Python3.9.x、Python3.10.x、Python3.11.x

2.3.1

v2.3.1-6.0.rc3

2.3.1.post2

2.4.0

v2.4.0-6.0.rc3

2.4.0

商发版：8.0.RC2

社区版：8.0.RC2.beta1

1.11.0

6.0.rc2

v1.11.0-6.0.rc2

1.11.0.post14

Python3.7.x（3.7.5及以上）、Python3.8.x、Python3.9.x、Python3.10.x

2.1.0

v2.1.0-6.0.rc2

2.1.0.post6

Python3.8.x、Python3.9.x、Python3.10.x

2.2.0

v2.2.0-6.0.rc2

2.2.0.post2

Python3.8.x、Python3.9.x、Python3.10.x

2.3.1

v2.3.1-6.0.rc2

2.3.1

Python3.8.x、Python3.9.x、Python3.10.x

商发版：8.0.RC1

社区版：8.0.RC1.beta1

1.11.0

6.0.rc1

v1.11.0-6.0.rc1

1.11.0.post11

Python3.7.x（3.7.5及以上）、Python3.8.x、Python3.9.x、Python3.10.x

2.1.0

v2.1.0-6.0.rc1

2.1.0.post3

Python3.8.x、Python3.9.x、Python3.10.x

2.2.0

v2.2.0-6.0.rc1

2.2.0

Python3.8.x、Python3.9.x、Python3.10.x

7.0.0

1.11.0

5.0.0

v1.11.0-5.0.0

1.11.0.post8

Python3.7.x（3.7.5及以上）、Python3.8.x、Python3.9.x、Python3.10.x

2.0.1

v2.0.1-5.0.0

2.0.1.post1

Python3.8.x、Python3.9.x、Python3.10.x

2.1.0

v2.1.0-5.0.0

2.1.0

Python3.8.x、Python3.9.x、Python3.10.x

CANN 8.0.RC3.20为补丁版本，仅适配Atlas A3 训练系列产品的集群训练场景。Ascend Extension for PyTorch 6.0.0版本配套使用CANN 8.0.RC3.20时，请单击CANN获取对应版本资料。

  • torch_npu.npu_mm_reduce_scatter_base、torch_npu.npu_all_gather_base_mm等部分接口不支持Atlas A3 训练系列产品，具体可参考《API 参考》。
  • Atlas A3 训练系列产品，torch.npu.config.allow_internal_format接口的配参默认值为False。
表2 组件间配套表

Ascend Extension for PyTorch版本

Ascend Extension for PyTorch代码分支名称

Ascend Extension for PyTorch安装包版本

MindSpeed

MindSpeed代码分支名称

MindSpeed-LLM

MindSpeed-MM

6.0.0

v2.1.0-6.0.0

2.1.0.post10

1.0.0

1.0.0_core_r0.6.0

1.0.0

1.0.0

1.0.0_core_r0.7.0

v2.3.1-6.0.0

2.3.1.post4

-

-

-

-

v2.4.0-6.0.0

2.4.0.post2

-

-

-

-

6.0.rc3

v2.1.0-6.0.rc3

2.1.0.post8

1.0.RC3

1.0.RC3_core_r0.6.0

1.0.RC3

1.0.RC3

1.0.RC3_core_r0.7.0

v2.3.1-6.0.rc3

2.3.1.post2

-

-

-

-

v2.4.0-6.0.rc3

2.4.0

-

-

-

-