From 1cb4e2d1d6e4e5a2e353356095a2680584ccb533 Mon Sep 17 00:00:00 2001 From: dilililiwhy Date: Tue, 4 Jun 2024 16:55:22 +0800 Subject: [PATCH] release 2.3.1rc1 --- README.md | 145 ++++++++++++++++++++------------------- README.zh.md | 137 ++++++++++++++++++------------------ ci/docker/ARM/Dockerfile | 9 ++- ci/docker/X86/Dockerfile | 9 ++- 4 files changed, 152 insertions(+), 148 deletions(-) diff --git a/README.md b/README.md index dd62e2aff04..09bce714db5 100644 --- a/README.md +++ b/README.md @@ -18,21 +18,21 @@ Install **PyTorch** through pip. **For Aarch64:** -```Python -pip3 install torch==2.1.0 +```Shell +pip3 install torch==2.3.1 ``` **For x86:** -```Python -pip3 install torch==2.1.0+cpu --index-url https://download.pytorch.org/whl/cpu +```Shell +pip3 install torch==2.3.1+cpu --index-url https://download.pytorch.org/whl/cpu ``` 2. **Install torch-npu dependencies** Run the following command to install dependencies. -```Python +```Shell pip3 install pyyaml pip3 install setuptools ``` @@ -41,17 +41,17 @@ If the installation fails, use the download link or visit the [PyTorch official | OS arch | Python version | link | | ------- | -------------- | ------------------------------------------------------------ | -| x86 | Python3.8 | [link](https://download.pytorch.org/whl/cpu/torch-2.1.0%2Bcpu-cp38-cp38-linux_x86_64.whl#sha256=9e5cfd931a65b38d222755a45dabb53b836be31bc620532bc66fee77e3ff67dc) | -| x86 | Python3.9 | [link](https://download.pytorch.org/whl/cpu/torch-2.1.0%2Bcpu-cp39-cp39-linux_x86_64.whl#sha256=86cc28df491fa84738affe752f9870791026565342f69e4ab63e5b935f00a495) | -| x86 | Python3.10 | [link](https://download.pytorch.org/whl/cpu/torch-2.1.0%2Bcpu-cp310-cp310-linux_x86_64.whl#sha256=5077921fc2b54e69a534f3a9c0b98493c79a5547c49d46f5e77e42da3610e011) | -| aarch64 | Python3.8 | [link](https://download.pytorch.org/whl/cpu/torch-2.1.0-cp38-cp38-manylinux_2_17_aarch64.manylinux2014_aarch64.whl#sha256=761822761fffaa1c18a62c5deb13abaa780862577d3eadc428f1daa632536905) | -| aarch64 | Python3.9 | [link](https://download.pytorch.org/whl/cpu/torch-2.1.0-cp39-cp39-manylinux_2_17_aarch64.manylinux2014_aarch64.whl#sha256=de7d63c6ecece118684415a3dbd4805af4a4c1ee1490cccf7405d8c240a481b4) | -| aarch64 | Python3.10 | [link](https://download.pytorch.org/whl/cpu/torch-2.1.0-cp310-cp310-manylinux_2_17_aarch64.manylinux2014_aarch64.whl#sha256=a04a0296d47f28960f51c18c5489a8c3472f624ec3b5bcc8e2096314df8c3342) | +| x86 | Python3.8 | [link](https://download.pytorch.org/whl/cpu/torch-2.3.1%2Bcpu-cp38-cp38-linux_x86_64.whl#sha256=cab92d5101e6db686c5525e04d87cedbcf3a556073d71d07fbe7d1ce09630ffb) | +| x86 | Python3.9 | [link](https://download.pytorch.org/whl/cpu/torch-2.3.1%2Bcpu-cp39-cp39-linux_x86_64.whl#sha256=a3cb8e61ba311cee1bb7463cbdcf3ebdfd071e2091e74c5785e3687eb02819f9) | +| x86 | Python3.10 | [link](https://download.pytorch.org/whl/cpu/torch-2.3.1%2Bcpu-cp310-cp310-linux_x86_64.whl#sha256=d679e21d871982b9234444331a26350902cfd2d5ca44ce6f49896af8b3a3087d) | +| aarch64 | Python3.8 | [link](https://download.pytorch.org/whl/cpu/torch-2.3.1-cp38-cp38-manylinux_2_17_aarch64.manylinux2014_aarch64.whl#sha256=3b7c1498f904f67eb1e331f2ebe8742771a2ce71b9ee9bc01de967257e881c7d) | +| aarch64 | Python3.9 | [link](https://download.pytorch.org/whl/cpu/torch-2.3.1-cp39-cp39-manylinux_2_17_aarch64.manylinux2014_aarch64.whl#sha256=db6bff4ba6273b59ae443de04b5adc36d6a40bb2898866133bff2d52f276eafe) | +| aarch64 | Python3.10 | [link](https://download.pytorch.org/whl/cpu/torch-2.3.1-cp310-cp310-manylinux_2_17_aarch64.manylinux2014_aarch64.whl#sha256=6544fdf29018668c0a6d4a1bcc955982c1ada70806281b010cba93bdcfbdcf22) | 3. **Install torch-npu** ``` -pip3 install torch-npu==2.1.0.post3 +pip3 install torch-npu==2.3.1rc1 ``` ### From Source @@ -61,7 +61,7 @@ In some special scenarios, users may need to compile **torch-npu** by themselves 1. **Clone torch-npu** ``` - git clone https://github.com/ascend/pytorch.git -b v2.1.0-6.0.rc1 --depth 1 + git clone https://github.com/ascend/pytorch.git -b v2.3.1 --depth 1 ``` 2. **Build Docker Image** @@ -120,53 +120,55 @@ Refer to [API of Ascend Extension for PyTorch](docs/api/torch_npu_apis.md) for m ## PyTorch and Python Version Matching Table | PyTorch Version | Python Version | -| ------------- | :----------------------------------------------------------- | -| PyTorch1.11.0 | Python3.7.x(>=3.7.5),Python3.8.x,Python3.9.x,Python3.10.x | -| PyTorch2.0.1 | Python3.8.x,Python3.9.x,Python3.10.x | -| PyTorch2.1.0 | Python3.8.x,Python3.9.x,Python3.10.x | +|-----------------|:----------------------------------------------------------| +| PyTorch1.11.0 | Python3.7.x(>=3.7.5),Python3.8.x,Python3.9.x,Python3.10.x | +| PyTorch2.1.0 | Python3.8.x,Python3.9.x,Python3.10.x | +| PyTorch2.2.0 | Python3.8.x,Python3.9.x,Python3.10.x | +| PyTorch2.3.1 | Python3.8.x,Python3.9.x,Python3.10.x | ## Ascend Auxiliary Software **PyTorch Extension** versions follow the naming convention `{PyTorch version}-{Ascend version}`, where the former represents the PyTorch version compatible with the **PyTorch Extension**, and the latter is used to match the CANN version. The detailed matching is as follows: -| CANN Version | Supported PyTorch Version | Supported Extension Version | Github Branch | AscendHub Image Version/Name([Link](https://ascendhub.huawei.com/#/detail/pytorch-modelzoo)) | -|----------------|--------------|-------------------|-------------------|----------------------| -| CANN 8.0.RC1 | 2.2.0 | 2.2.0 | v2.2.0-6.0.rc1 | - | -| | 2.1.0 | 2.1.0.post4 | v2.1.0-6.0.rc1 | - | -| | 1.11.0 | 1.11.0.post11 | v1.11.0-6.0.rc1 | - | -| CANN 7.0.0 | 2.1.0 | 2.1.0 | v2.1.0-5.0.0 | - | -| | 2.0.1 | 2.0.1.post1 | v2.0.1-5.0.0 | - | -| | 1.11.0 | 1.11.0.post8 | v1.11.0-5.0.0 | - | -| CANN 7.0.RC1 | 2.1.0 | 2.1.0.rc1 | v2.1.0-5.0.rc3 | - | -| | 2.0.1 | 2.0.1 | v2.0.1-5.0.rc3 | - | -| | 1.11.0 | 1.11.0.post4 | v1.11.0-5.0.rc3 | - | -| CANN 6.3.RC3.1 | 1.11.0 | 1.11.0.post3 | v1.11.0-5.0.rc2.2 | - | -| CANN 6.3.RC3 | 1.11.0 | 1.11.0.post2 | v1.11.0-5.0.rc2.1 | - | -| CANN 6.3.RC2 | 2.0.1 | 2.0.1.rc1 | v2.0.1-5.0.rc2 | - | -| | 1.11.0 | 1.11.0.post1 | v1.11.0-5.0.rc2 | 23.0.RC1-1.11.0 | -| | 1.8.1 | 1.8.1.post2 | v1.8.1-5.0.rc2 | 23.0.RC1-1.8.1 | -| CANN 6.3.RC1 | 1.11.0 | 1.11.0 | v1.11.0-5.0.rc1 | - | -| | 1.8.1 | 1.8.1.post1 | v1.8.1-5.0.rc1 | - | -| CANN 6.0.1 | 1.5.0 | 1.5.0.post8 | v1.5.0-3.0.0 | 22.0.0 | -| | 1.8.1 | 1.8.1 | v1.8.1-3.0.0 | 22.0.0-1.8.1 | -| | 1.11.0 | 1.11.0.rc2（beta) | v1.11.0-3.0.0 | - | -| CANN 6.0.RC1 | 1.5.0 | 1.5.0.post7 | v1.5.0-3.0.rc3 | 22.0.RC3 | -| | 1.8.1 | 1.8.1.rc3 | v1.8.1-3.0.rc3 | 22.0.RC3-1.8.1 | -| | 1.11.0 | 1.11.0.rc1（beta) | v1.11.0-3.0.rc3 | - | -| CANN 5.1.RC2 | 1.5.0 | 1.5.0.post6 | v1.5.0-3.0.rc2 | 22.0.RC2 | -| | 1.8.1 | 1.8.1.rc2 | v1.8.1-3.0.rc2 | 22.0.RC2-1.8.1 | -| CANN 5.1.RC1 | 1.5.0 | 1.5.0.post5 | v1.5.0-3.0.rc1 | 22.0.RC1 | -| | 1.8.1 | 1.8.1.rc1 | v1.8.1-3.0.rc1 | - | -| CANN 5.0.4 | 1.5.0 | 1.5.0.post4 | 2.0.4.tr5 | 21.0.4 | -| CANN 5.0.3 | 1.8.1 | 1.5.0.post3 | 2.0.3.tr5 | 21.0.3 | -| CANN 5.0.2 | 1.5.0 | 1.5.0.post2 | 2.0.2.tr5 | 21.0.2 | +| CANN Version | Supported PyTorch Version | Supported Extension Version | Github Branch | AscendHub Image Version/Name([Link](https://ascendhub.huawei.com/#/detail/pytorch-modelzoo)) | +|-----------------------|---------------------------|-----------------------------|-------------------|----------------------------------------------------------------------------------------------| +| CANN 8.0.RC2.alpha002 | 2.3.1 | 2.3.1rc1 | v2.3.1 | - | +| CANN 8.0.RC1 | 2.2.0 | 2.2.0 | v2.2.0-6.0.rc1 | - | +| | 2.1.0 | 2.1.0.post4 | v2.1.0-6.0.rc1 | - | +| | 1.11.0 | 1.11.0.post11 | v1.11.0-6.0.rc1 | - | +| CANN 7.0.0 | 2.1.0 | 2.1.0 | v2.1.0-5.0.0 | - | +| | 2.0.1 | 2.0.1.post1 | v2.0.1-5.0.0 | - | +| | 1.11.0 | 1.11.0.post8 | v1.11.0-5.0.0 | - | +| CANN 7.0.RC1 | 2.1.0 | 2.1.0.rc1 | v2.1.0-5.0.rc3 | - | +| | 2.0.1 | 2.0.1 | v2.0.1-5.0.rc3 | - | +| | 1.11.0 | 1.11.0.post4 | v1.11.0-5.0.rc3 | - | +| CANN 6.3.RC3.1 | 1.11.0 | 1.11.0.post3 | v1.11.0-5.0.rc2.2 | - | +| CANN 6.3.RC3 | 1.11.0 | 1.11.0.post2 | v1.11.0-5.0.rc2.1 | - | +| CANN 6.3.RC2 | 2.0.1 | 2.0.1.rc1 | v2.0.1-5.0.rc2 | - | +| | 1.11.0 | 1.11.0.post1 | v1.11.0-5.0.rc2 | 23.0.RC1-1.11.0 | +| | 1.8.1 | 1.8.1.post2 | v1.8.1-5.0.rc2 | 23.0.RC1-1.8.1 | +| CANN 6.3.RC1 | 1.11.0 | 1.11.0 | v1.11.0-5.0.rc1 | - | +| | 1.8.1 | 1.8.1.post1 | v1.8.1-5.0.rc1 | - | +| CANN 6.0.1 | 1.5.0 | 1.5.0.post8 | v1.5.0-3.0.0 | 22.0.0 | +| | 1.8.1 | 1.8.1 | v1.8.1-3.0.0 | 22.0.0-1.8.1 | +| | 1.11.0 | 1.11.0.rc2（beta) | v1.11.0-3.0.0 | - | +| CANN 6.0.RC1 | 1.5.0 | 1.5.0.post7 | v1.5.0-3.0.rc3 | 22.0.RC3 | +| | 1.8.1 | 1.8.1.rc3 | v1.8.1-3.0.rc3 | 22.0.RC3-1.8.1 | +| | 1.11.0 | 1.11.0.rc1（beta) | v1.11.0-3.0.rc3 | - | +| CANN 5.1.RC2 | 1.5.0 | 1.5.0.post6 | v1.5.0-3.0.rc2 | 22.0.RC2 | +| | 1.8.1 | 1.8.1.rc2 | v1.8.1-3.0.rc2 | 22.0.RC2-1.8.1 | +| CANN 5.1.RC1 | 1.5.0 | 1.5.0.post5 | v1.5.0-3.0.rc1 | 22.0.RC1 | +| | 1.8.1 | 1.8.1.rc1 | v1.8.1-3.0.rc1 | - | +| CANN 5.0.4 | 1.5.0 | 1.5.0.post4 | 2.0.4.tr5 | 21.0.4 | +| CANN 5.0.3 | 1.8.1 | 1.5.0.post3 | 2.0.3.tr5 | 21.0.3 | +| CANN 5.0.2 | 1.5.0 | 1.5.0.post2 | 2.0.2.tr5 | 21.0.2 | ## Pipeline Status Due to the asynchronous development mechanism of upstream and downstream, incompatible modifications in upstream may cause some functions of **torch_npu** to be unavailable (only upstream and downstream development branches are involved, excluding stable branches). Therefore, we built a set of daily tasks that make it easy to detect relevant issues in time and fix them within 48 hours (under normal circumstances), providing users with the latest features and stable quality. -| **OS** | **CANN Version(Docker Image)** | **Upstream Branch** | **Downstream Branch** | **Period** | **Status** | -| :---: | :---: | :---: | :---: | :---: | :---: | +| **OS** | **CANN Version(Docker Image)** | **Upstream Branch** | **Downstream Branch** | **Period** | **Status** | +|:-------------------:|:-------------------------------------------------------:|:----------------------------------------------------:|:-------------------------------------------------------:|:--------------:|:-------------------------------------------------------------------------------------------------------------------------------------------------------------:| | openEuler 22.03 SP2 | [CANN 7.1](https://hub.docker.com/r/ascendai/cann/tags) | [main](https://github.com/pytorch/pytorch/tree/main) | [master](https://github.com/Ascend/pytorch/tree/master) | UTC 1200 daily | [![Ascend NPU](https://github.com/Ascend/pytorch/actions/workflows/periodic.yml/badge.svg)](https://github.com/Ascend/pytorch/actions/workflows/periodic.yml) | ## Suggestions and Communication @@ -177,35 +179,36 @@ Everyone is welcome to contribute to the community. If you have any questions or The version branches of AscendPyTorch have the following maintenance phases: -| **Status** | **Duration** | **Description** | -|-------------------|--------------|--------------------------------------------------------------------------------------------------------------------------------| -| Planning | 1-3 months | Plan features. | -| Development | 3 months | Develop features. | -| Maintained | 6-12 months | Allow the incorporation of all resolved issues and release the version, Different versions of PyTorch adopt varying support policies. The maintenance periods for Regular Releases and Long-Term Support versions are 6 months and 12 months, respectively. | -| Unmaintained | 0-3 months | Allow the incorporation of all resolved issues. No dedicated maintenance personnel are available. No version will be released. | -| End Of Life (EOL) | N/A | Do not accept any modification to a branch. | +| **Status** | **Duration** | **Description** | +|-------------------|--------------|-------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------| +| Planning | 1-3 months | Plan features. | +| Development | 3 months | Develop features. | +| Maintained | 6-12 months | Allow the incorporation of all resolved issues and release the version, Different versions of PyTorch adopt varying support policies. The maintenance periods for Regular Releases and Long-Term Support versions are 6 months and 12 months, respectively. | +| Unmaintained | 0-3 months | Allow the incorporation of all resolved issues. No dedicated maintenance personnel are available. No version will be released. | +| End Of Life (EOL) | N/A | Do not accept any modification to a branch. | ## PyTorch Maintenance Policies -| **PyTorch** | **Maintenance Policies** | **Status** | **Launch Date** | **Subsequent Status** | **EOL Date** | -|-----------|--------------------|--------------|------------|-----------------|-----------| -| 2.2.0 | Regular Release | Maintained | 2024/04/01 | Unmaintained 2024-10-15 estimated | | -| 2.1.0 | Long Term Support | Maintained | 2023/10/15 | Unmaintained 2024-10-15 estimated | | -| 2.0.1 | Regular Release | EOL | 2023/7/19 | | 2024/3/14 | -| 1.11.0 | Long Term Support | Maintained | 2023/4/19 | Unmaintained 2024-4-19 estimated | | -| 1.8.1 | Long Term Support | EOL | 2022/4/10 | | 2023/4/10 | -| 1.5.0 | Long Term Support | EOL | 2021/7/29 | | 2022/7/29 | +| **PyTorch** | **Maintenance Policies** | **Status** | **Launch Date** | **Subsequent Status** | **EOL Date** | +|-------------|--------------------------|------------|-----------------|-----------------------------------|--------------| +| 2.3.1 | Regular Release | Maintained | 2024/06/06 | Unmaintained 2024-12-06 estimated | | +| 2.2.0 | Regular Release | Maintained | 2024/04/01 | Unmaintained 2024-10-15 estimated | | +| 2.1.0 | Long Term Support | Maintained | 2023/10/15 | Unmaintained 2024-10-15 estimated | | +| 2.0.1 | Regular Release | EOL | 2023/7/19 | | 2024/3/14 | +| 1.11.0 | Long Term Support | Maintained | 2023/4/19 | Unmaintained 2024-4-19 estimated | | +| 1.8.1 | Long Term Support | EOL | 2022/4/10 | | 2023/4/10 | +| 1.5.0 | Long Term Support | EOL | 2021/7/29 | | 2022/7/29 | ## Reference Documents For more detailed information on installation guides, model migration, training/inference tutorials, and API lists, please refer to the [Ascend Extension for PyTorch on the HiAI Community](https://www.hiascend.com/software/ai-frameworks/commercial). -| Document Name | Document Link | -| -------------------------------- | ------------------------------------------------------------ | -| Installation Guide | [link](https://www.hiascend.com/document/detail/zh/Pytorch/60RC1/quickstart/instg) | -| Network Model Migration and Training | [link](https://www.hiascend.com/document/detail/zh/Pytorch/60RC1/ptmoddevg/trainingmigrguide/PT_LMTMOG_0003.html) | -| Operator Adaptation | [link](https://www.hiascend.com/document/detail/zh/canncommercial/700/operatordev/tbeaicpudevg/atlasopdev_10_0086.html) | -| API List (PyTorch and Custom Interfaces) | [link](https://www.hiascend.com/document/detail/zh/Pytorch/60RC1/apiref/apilist/ptaoplist_000002.html) | +| Document Name | Document Link | +|------------------------------------------|-------------------------------------------------------------------------------------------------------------------------| +| Installation Guide | [link](https://www.hiascend.com/document/detail/zh/Pytorch/60RC1/quickstart/instg) | +| Network Model Migration and Training | [link](https://www.hiascend.com/document/detail/zh/Pytorch/60RC1/ptmoddevg/trainingmigrguide/PT_LMTMOG_0003.html) | +| Operator Adaptation | [link](https://www.hiascend.com/document/detail/zh/canncommercial/700/operatordev/tbeaicpudevg/atlasopdev_10_0086.html) | +| API List (PyTorch and Custom Interfaces) | [link](https://www.hiascend.com/document/detail/zh/Pytorch/60RC1/apiref/apilist/ptaoplist_000002.html) | ## License diff --git a/README.zh.md b/README.zh.md index 141d8926cac..540d0b80c82 100644 --- a/README.zh.md +++ b/README.zh.md @@ -18,32 +18,32 @@ **aarch64:** -```Python -pip3 install torch==2.1.0 +```Shell +pip3 install torch==2.3.1 ``` **x86:** -```Python -pip3 install torch==2.1.0+cpu --index-url https://download.pytorch.org/whl/cpu +```Shell +pip3 install torch==2.3.1+cpu --index-url https://download.pytorch.org/whl/cpu ``` 若使用pip命令安装失败，请使用下载链接或进入[PyTorch官方网站](https://pytorch.org/)进行查询下载对应版本。 | 架构 | Python版本 | 下载链接 | | ------- | ---------- | ------------------------------------------------------------ | -| x86 | Python3.8 | [下载链接](https://download.pytorch.org/whl/cpu/torch-2.1.0%2Bcpu-cp38-cp38-linux_x86_64.whl#sha256=9e5cfd931a65b38d222755a45dabb53b836be31bc620532bc66fee77e3ff67dc) | -| x86 | Python3.9 | [下载链接](https://download.pytorch.org/whl/cpu/torch-2.1.0%2Bcpu-cp39-cp39-linux_x86_64.whl#sha256=86cc28df491fa84738affe752f9870791026565342f69e4ab63e5b935f00a495) | -| x86 | Python3.10 | [下载链接](https://download.pytorch.org/whl/cpu/torch-2.1.0%2Bcpu-cp310-cp310-linux_x86_64.whl#sha256=5077921fc2b54e69a534f3a9c0b98493c79a5547c49d46f5e77e42da3610e011) | -| aarch64 | Python3.8 | [下载链接](https://download.pytorch.org/whl/cpu/torch-2.1.0-cp38-cp38-manylinux_2_17_aarch64.manylinux2014_aarch64.whl#sha256=761822761fffaa1c18a62c5deb13abaa780862577d3eadc428f1daa632536905) | -| aarch64 | Python3.9 | [下载链接](https://download.pytorch.org/whl/cpu/torch-2.1.0-cp39-cp39-manylinux_2_17_aarch64.manylinux2014_aarch64.whl#sha256=de7d63c6ecece118684415a3dbd4805af4a4c1ee1490cccf7405d8c240a481b4) | -| aarch64 | Python3.10 | [下载链接](https://download.pytorch.org/whl/cpu/torch-2.1.0-cp310-cp310-manylinux_2_17_aarch64.manylinux2014_aarch64.whl#sha256=a04a0296d47f28960f51c18c5489a8c3472f624ec3b5bcc8e2096314df8c3342) | +| x86 | Python3.8 | [下载链接](https://download.pytorch.org/whl/cpu/torch-2.3.1%2Bcpu-cp38-cp38-linux_x86_64.whl#sha256=cab92d5101e6db686c5525e04d87cedbcf3a556073d71d07fbe7d1ce09630ffb) | +| x86 | Python3.9 | [下载链接](https://download.pytorch.org/whl/cpu/torch-2.3.1%2Bcpu-cp39-cp39-linux_x86_64.whl#sha256=a3cb8e61ba311cee1bb7463cbdcf3ebdfd071e2091e74c5785e3687eb02819f9) | +| x86 | Python3.10 | [下载链接](https://download.pytorch.org/whl/cpu/torch-2.3.1%2Bcpu-cp310-cp310-linux_x86_64.whl#sha256=d679e21d871982b9234444331a26350902cfd2d5ca44ce6f49896af8b3a3087d) | +| aarch64 | Python3.8 | [下载链接](https://download.pytorch.org/whl/cpu/torch-2.3.1-cp38-cp38-manylinux_2_17_aarch64.manylinux2014_aarch64.whl#sha256=3b7c1498f904f67eb1e331f2ebe8742771a2ce71b9ee9bc01de967257e881c7d) | +| aarch64 | Python3.9 | [下载链接](https://download.pytorch.org/whl/cpu/torch-2.3.1-cp39-cp39-manylinux_2_17_aarch64.manylinux2014_aarch64.whl#sha256=db6bff4ba6273b59ae443de04b5adc36d6a40bb2898866133bff2d52f276eafe) | +| aarch64 | Python3.10 | [下载链接](https://download.pytorch.org/whl/cpu/torch-2.3.1-cp310-cp310-manylinux_2_17_aarch64.manylinux2014_aarch64.whl#sha256=6544fdf29018668c0a6d4a1bcc955982c1ada70806281b010cba93bdcfbdcf22) | 2. **安装torch_npu依赖** 运行以下命令安装依赖。 -```Python +```Shell pip3 install pyyaml pip3 install setuptools ``` @@ -51,7 +51,7 @@ pip3 install setuptools 3. **安装torch_npu** ``` -pip3 install torch-npu==2.1.0.post3 +pip3 install torch-npu==2.3.1rc1 ``` 如需要保存安装日志，可在pip3 install命令后面加上参数 `--log `，并对您指定的目录``做好权限管控。 @@ -62,7 +62,7 @@ pip3 install torch-npu==2.1.0.post3 1. **克隆torch_npu代码仓** ``` - git clone https://gitee.com/ascend/pytorch.git -b v2.1.0-6.0.rc1 --depth 1 + git clone https://gitee.com/ascend/pytorch.git -b v2.3.1 --depth 1 ``` 2. **构建镜像** @@ -128,47 +128,49 @@ print(z) ## PyTorch与Python版本配套表 -| PyTorch版本 | Python版本 | -| ------------- | :----------------------------------------------------------- | +| PyTorch版本 | Python版本 | +|---------------|:-------------------------------------------------------------| | PyTorch1.11.0 | Python3.7.x(>=3.7.5), Python3.8.x, Python3.9.x, Python3.10.x | | PyTorch2.1.0 | Python3.8.x, Python3.9.x, Python3.10.x | | PyTorch2.2.0 | Python3.8.x, Python3.9.x, Python3.10.x | +| PyTorch2.3.1 | Python3.8.x, Python3.9.x, Python3.10.x | ## 昇腾辅助软件 **PyTorch Extension**版本号采用`{PyTorch版本}-{昇腾版本}`命名规则，前者为**PyTorch Extension**匹配的PyTorch版本，后者用于匹配CANN版本，详细匹配如下： -| CANN版本 | 支持的PyTorch版本 | 支持的Extension版本 | Gitee分支 | AscendHub镜像版本/名称([链接](https://ascendhub.huawei.com/#/detail/pytorch-modelzoo)) | -|----------------|--------------|-------------------|-------------------|----------------------| -| CANN 8.0.RC1 | 2.2.0 | 2.2.0 | v2.2.0-6.0.rc1 | - | -| | 2.1.0 | 2.1.0.post4 | v2.1.0-6.0.rc1 | - | -| | 1.11.0 | 1.11.0.post11 | v1.11.0-6.0.rc1 | - | -| CANN 7.0.0 | 2.1.0 | 2.1.0 | v2.1.0-5.0.0 | - | -| | 2.0.1 | 2.0.1.post1 | v2.0.1-5.0.0 | - | -| | 1.11.0 | 1.11.0.post8 | v1.11.0-5.0.0 | - | -| CANN 7.0.RC1 | 2.1.0 | 2.1.0.rc1 | v2.1.0-5.0.rc3 | - | -| | 2.0.1 | 2.0.1 | v2.0.1-5.0.rc3 | - | -| | 1.11.0 | 1.11.0.post4 | v1.11.0-5.0.rc3 | - | -| CANN 6.3.RC3.1 | 1.11.0 | 1.11.0.post3 | v1.11.0-5.0.rc2.2 | - | -| CANN 6.3.RC3 | 1.11.0 | 1.11.0.post2 | v1.11.0-5.0.rc2.1 | - | -| CANN 6.3.RC2 | 2.0.1 | 2.0.1.rc1 | v2.0.1-5.0.rc2 | - | -| | 1.11.0 | 1.11.0.post1 | v1.11.0-5.0.rc2 | 23.0.RC1-1.11.0 | -| | 1.8.1 | 1.8.1.post2 | v1.8.1-5.0.rc2 | 23.0.RC1-1.8.1 | -| CANN 6.3.RC1 | 1.11.0 | 1.11.0 | v1.11.0-5.0.rc1 | - | -| | 1.8.1 | 1.8.1.post1 | v1.8.1-5.0.rc1 | - | -| CANN 6.0.1 | 1.5.0 | 1.5.0.post8 | v1.5.0-3.0.0 | 22.0.0 | -| | 1.8.1 | 1.8.1 | v1.8.1-3.0.0 | 22.0.0-1.8.1 | -| | 1.11.0 | 1.11.0.rc2（beta) | v1.11.0-3.0.0 | - | -| CANN 6.0.RC1 | 1.5.0 | 1.5.0.post7 | v1.5.0-3.0.rc3 | 22.0.RC3 | -| | 1.8.1 | 1.8.1.rc3 | v1.8.1-3.0.rc3 | 22.0.RC3-1.8.1 | -| | 1.11.0 | 1.11.0.rc1（beta) | v1.11.0-3.0.rc3 | - | -| CANN 5.1.RC2 | 1.5.0 | 1.5.0.post6 | v1.5.0-3.0.rc2 | 22.0.RC2 | -| | 1.8.1 | 1.8.1.rc2 | v1.8.1-3.0.rc2 | 22.0.RC2-1.8.1 | -| CANN 5.1.RC1 | 1.5.0 | 1.5.0.post5 | v1.5.0-3.0.rc1 | 22.0.RC1 | -| | 1.8.1 | 1.8.1.rc1 | v1.8.1-3.0.rc1 | - | -| CANN 5.0.4 | 1.5.0 | 1.5.0.post4 | 2.0.4.tr5 | 21.0.4 | -| CANN 5.0.3 | 1.8.1 | 1.5.0.post3 | 2.0.3.tr5 | 21.0.3 | -| CANN 5.0.2 | 1.5.0 | 1.5.0.post2 | 2.0.2.tr5 | 21.0.2 | +| CANN版本 | 支持的PyTorch版本 | 支持的Extension版本 | Gitee分支 | AscendHub镜像版本/名称([链接](https://ascendhub.huawei.com/#/detail/pytorch-modelzoo)) | +|-----------------------|--------------|------------------|-------------------|--------------------------------------------------------------------------------| +| CANN 8.0.RC2.alpha002 | 2.3.1 | 2.3.1rc1 | v2.3.1 | - | +| CANN 8.0.RC1 | 2.2.0 | 2.2.0 | v2.2.0-6.0.rc1 | - | +| | 2.1.0 | 2.1.0.post4 | v2.1.0-6.0.rc1 | - | +| | 1.11.0 | 1.11.0.post11 | v1.11.0-6.0.rc1 | - | +| CANN 7.0.0 | 2.1.0 | 2.1.0 | v2.1.0-5.0.0 | - | +| | 2.0.1 | 2.0.1.post1 | v2.0.1-5.0.0 | - | +| | 1.11.0 | 1.11.0.post8 | v1.11.0-5.0.0 | - | +| CANN 7.0.RC1 | 2.1.0 | 2.1.0.rc1 | v2.1.0-5.0.rc3 | - | +| | 2.0.1 | 2.0.1 | v2.0.1-5.0.rc3 | - | +| | 1.11.0 | 1.11.0.post4 | v1.11.0-5.0.rc3 | - | +| CANN 6.3.RC3.1 | 1.11.0 | 1.11.0.post3 | v1.11.0-5.0.rc2.2 | - | +| CANN 6.3.RC3 | 1.11.0 | 1.11.0.post2 | v1.11.0-5.0.rc2.1 | - | +| CANN 6.3.RC2 | 2.0.1 | 2.0.1.rc1 | v2.0.1-5.0.rc2 | - | +| | 1.11.0 | 1.11.0.post1 | v1.11.0-5.0.rc2 | 23.0.RC1-1.11.0 | +| | 1.8.1 | 1.8.1.post2 | v1.8.1-5.0.rc2 | 23.0.RC1-1.8.1 | +| CANN 6.3.RC1 | 1.11.0 | 1.11.0 | v1.11.0-5.0.rc1 | - | +| | 1.8.1 | 1.8.1.post1 | v1.8.1-5.0.rc1 | - | +| CANN 6.0.1 | 1.5.0 | 1.5.0.post8 | v1.5.0-3.0.0 | 22.0.0 | +| | 1.8.1 | 1.8.1 | v1.8.1-3.0.0 | 22.0.0-1.8.1 | +| | 1.11.0 | 1.11.0.rc2（beta) | v1.11.0-3.0.0 | - | +| CANN 6.0.RC1 | 1.5.0 | 1.5.0.post7 | v1.5.0-3.0.rc3 | 22.0.RC3 | +| | 1.8.1 | 1.8.1.rc3 | v1.8.1-3.0.rc3 | 22.0.RC3-1.8.1 | +| | 1.11.0 | 1.11.0.rc1（beta) | v1.11.0-3.0.rc3 | - | +| CANN 5.1.RC2 | 1.5.0 | 1.5.0.post6 | v1.5.0-3.0.rc2 | 22.0.RC2 | +| | 1.8.1 | 1.8.1.rc2 | v1.8.1-3.0.rc2 | 22.0.RC2-1.8.1 | +| CANN 5.1.RC1 | 1.5.0 | 1.5.0.post5 | v1.5.0-3.0.rc1 | 22.0.RC1 | +| | 1.8.1 | 1.8.1.rc1 | v1.8.1-3.0.rc1 | - | +| CANN 5.0.4 | 1.5.0 | 1.5.0.post4 | 2.0.4.tr5 | 21.0.4 | +| CANN 5.0.3 | 1.8.1 | 1.5.0.post3 | 2.0.3.tr5 | 21.0.3 | +| CANN 5.0.2 | 1.5.0 | 1.5.0.post2 | 2.0.2.tr5 | 21.0.2 | ## 建议与交流 @@ -178,24 +180,25 @@ print(z) AscendPyTorch版本分支的维护阶段如下： -| **状态** | **时间** | **说明** | -| ------------------- | -------- | ------------------------------------------------ | -| 计划 | 1—3 个月 | 计划特性 | -| 开发 | 3 个月 | 开发特性 | -| 维护 | 6-12 个月| 合入所有已解决的问题并发布版本，针对不同的PyTorch版本采取不同的维护策略，常规版本和长期支持版本维护周期分别为6个月和12个月 | -| 无维护 | 0—3 个月 | 合入所有已解决的问题，无专职维护人员，无版本发布 | -| 生命周期终止（EOL） | N/A | 分支不再接受任何修改 | +| **状态** | **时间** | **说明** | +|-------------|---------|--------------------------------------------------------------------| +| 计划 | 1—3 个月 | 计划特性 | +| 开发 | 3 个月 | 开发特性 | +| 维护 | 6-12 个月 | 合入所有已解决的问题并发布版本，针对不同的PyTorch版本采取不同的维护策略，常规版本和长期支持版本维护周期分别为6个月和12个月 | +| 无维护 | 0—3 个月 | 合入所有已解决的问题，无专职维护人员，无版本发布 | +| 生命周期终止（EOL） | N/A | 分支不再接受任何修改 | ## PyTorch版本维护策略 -| **PyTorch版本** | **维护策略** | **当前状态** | **发布时间** | **后续状态** | **EOL日期** | -|-----------|-----------|--------|------------|-----------------------|-----------| -| 2.2.0 | 常规版本 | 维护 | 2024/04/01 | 预计2024/10/15起无维护 | | -| 2.1.0 | 长期支持 | 维护 | 2023/10/15 | 预计2024/10/15起无维护 | | -| 2.0.1 | 常规版本 | EOL | 2023/7/19 | | 2024/3/14 | -| 1.11.0 | 长期支持 | 维护 | 2023/4/19 | 预计2024/4/19起无维护 | | -| 1.8.1 | 长期支持 | EOL | 2022/4/10 | | 2023/4/10 | -| 1.5.0 | 长期支持 | EOL | 2021/7/29 | | 2022/7/29 | +| **PyTorch版本** | **维护策略** | **当前状态** | **发布时间** | **后续状态** | **EOL日期** | +|---------------|----------|----------|------------|------------------|-----------| +| 2.3.1 | 常规版本 | 维护 | 2024/06/06 | 预计2024/12/06起无维护 | | +| 2.2.0 | 常规版本 | 维护 | 2024/04/01 | 预计2024/10/15起无维护 | | +| 2.1.0 | 长期支持 | 维护 | 2023/10/15 | 预计2024/10/15起无维护 | | +| 2.0.1 | 常规版本 | EOL | 2023/7/19 | | 2024/3/14 | +| 1.11.0 | 长期支持 | 维护 | 2023/4/19 | 预计2024/4/19起无维护 | | +| 1.8.1 | 长期支持 | EOL | 2022/4/10 | | 2023/4/10 | +| 1.5.0 | 长期支持 | EOL | 2021/7/29 | | 2022/7/29 | ## 安全声明 @@ -205,12 +208,12 @@ AscendPyTorch版本分支的维护阶段如下： 有关安装指南、模型迁移和训练/推理教程和API列表等更多详细信息，请参考[昇腾社区Ascend Extension for PyTorch](https://www.hiascend.com/software/ai-frameworks/commercial)。 -| 文档名称 | 文档链接 | -| -------------------------- | ------------------------------------------------------------ | -| 安装指南 | [参考链接](https://www.hiascend.com/document/detail/zh/Pytorch/60RC1/quickstart/instg) | -| 网络模型迁移和训练 | [参考链接](https://www.hiascend.com/document/detail/zh/Pytorch/60RC1/ptmoddevg/trainingmigrguide/PT_LMTMOG_0003.html) | -| 算子适配 | [参考链接](https://www.hiascend.com/document/detail/zh/canncommercial/700/operatordev/tbeaicpudevg/atlasopdev_10_0086.html) | -| API清单（PyTorch原生接口与自定义接口） | [参考链接](https://www.hiascend.com/document/detail/zh/Pytorch/60RC1/apiref/apilist/ptaoplist_000002.html) | +| 文档名称 | 文档链接 | +|--------------------------|-------------------------------------------------------------------------------------------------------------------------| +| 安装指南 | [参考链接](https://www.hiascend.com/document/detail/zh/Pytorch/60RC1/quickstart/instg) | +| 网络模型迁移和训练 | [参考链接](https://www.hiascend.com/document/detail/zh/Pytorch/60RC1/ptmoddevg/trainingmigrguide/PT_LMTMOG_0003.html) | +| 算子适配 | [参考链接](https://www.hiascend.com/document/detail/zh/canncommercial/700/operatordev/tbeaicpudevg/atlasopdev_10_0086.html) | +| API清单（PyTorch原生接口与自定义接口） | [参考链接](https://www.hiascend.com/document/detail/zh/Pytorch/60RC1/apiref/apilist/ptaoplist_000002.html) | ## 许可证 diff --git a/ci/docker/ARM/Dockerfile b/ci/docker/ARM/Dockerfile index 2d3edbc4b89..cf842165005 100644 --- a/ci/docker/ARM/Dockerfile +++ b/ci/docker/ARM/Dockerfile @@ -17,16 +17,15 @@ RUN mkdir /root/.pip \ # Install pip package(build) RUN pip3.8 install pyyaml \ - && pip3.8 install torch==2.1.0 \ - && pip3.8 install numpy==1.21.3 + && pip3.8 install torch==2.3.1 \ + && pip3.8 install numpy==1.21.3 RUN pip3.9 install pyyaml \ - && pip3.9 install torch==2.1.0 \ + && pip3.9 install torch==2.3.1 \ && pip3.9 install numpy==1.21.3 RUN pip3.10 install pyyaml \ - && pip3.10 install torch==2.1.0 \ + && pip3.10 install torch==2.3.1 \ && pip3.10 install numpy==1.21.3 WORKDIR /home - diff --git a/ci/docker/X86/Dockerfile b/ci/docker/X86/Dockerfile index b44ea928bc4..0c08051e0d3 100644 --- a/ci/docker/X86/Dockerfile +++ b/ci/docker/X86/Dockerfile @@ -21,17 +21,16 @@ RUN mkdir /root/.pip \ # Install pip package(build) RUN pip3.8 install pyyaml \ - && pip3.8 install torch==2.1.0+cpu --index-url https://download.pytorch.org/whl/cpu \ - && pip3.8 install numpy==1.21.3 + && pip3.8 install torch==2.3.1+cpu --index-url https://download.pytorch.org/whl/cpu \ + && pip3.8 install numpy==1.21.3 RUN pip3.9 install pyyaml \ - && pip3.9 install torch==2.1.0+cpu --index-url https://download.pytorch.org/whl/cpu \ + && pip3.9 install torch==2.3.1+cpu --index-url https://download.pytorch.org/whl/cpu \ && pip3.9 install numpy==1.21.3 RUN pip3.10 install pyyaml \ - && pip3.10 install torch==2.1.0+cpu --index-url https://download.pytorch.org/whl/cpu \ + && pip3.10 install torch==2.3.1+cpu --index-url https://download.pytorch.org/whl/cpu \ && pip3.10 install numpy==1.21.3 WORKDIR /home - -- Gitee