Skip to content

Commit

Permalink
update CANN and PyTorch version
Browse files Browse the repository at this point in the history
  • Loading branch information
hipudding committed Jul 31, 2024
1 parent 849862e commit 525387a
Show file tree
Hide file tree
Showing 4 changed files with 48 additions and 7 deletions.
Binary file modified _static/images/pytorch_wechat.jpg
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
48 changes: 43 additions & 5 deletions _static/package_info.js
Original file line number Diff line number Diff line change
@@ -1,16 +1,16 @@
const package_info = {
'8.0.RC2.alpha002': {
'8.0.RC3': {
x86_64: {
url: 'https://ascend-repo.obs.cn-east-2.myhuaweicloud.com/Milan-ASL/Milan-ASL V100R001C18SPC702/Ascend-cann-toolkit_8.0.RC2.alpha002_linux-x86_64.run'
url: 'https://ascend-repo.obs.cn-east-2.myhuaweicloud.com/Milan-ASL/Milan-ASL V100R001C19SPC802/Ascend-cann-toolkit_8.0.RC3.alpha001_linux-x86_64.run'
},
aarch64: {
url: 'https://ascend-repo.obs.cn-east-2.myhuaweicloud.com/Milan-ASL/Milan-ASL V100R001C18SPC702/Ascend-cann-toolkit_8.0.RC2.alpha002_linux-aarch64.run'
url: 'https://ascend-repo.obs.cn-east-2.myhuaweicloud.com/Milan-ASL/Milan-ASL V100R001C19SPC802/Ascend-cann-toolkit_8.0.RC3.alpha001_linux-aarch64.run'
},
'910b': {
driver_version: '23.0.3',
firmware_version: '7.1.0.5.220',
firmware_url: 'https://ascend-repo.obs.cn-east-2.myhuaweicloud.com/Ascend HDK/Ascend HDK 23.0.3/Ascend-hdk-910b-npu-firmware_7.1.0.5.220.run',
kernel_url: 'https://ascend-repo.obs.cn-east-2.myhuaweicloud.com/Milan-ASL/Milan-ASL V100R001C18SPC702/Ascend-cann-kernels-910b_8.0.RC2.alpha002_linux.run',
kernel_url: 'https://ascend-repo.obs.cn-east-2.myhuaweicloud.com/Milan-ASL/Milan-ASL V100R001C19SPC802/Ascend-cann-kernels-910b_8.0.RC3.alpha001_linux.run',
x86_64: {
driver_url: 'https://ascend-repo.obs.cn-east-2.myhuaweicloud.com/Ascend HDK/Ascend HDK 23.0.3/Ascend-hdk-910b-npu-driver_23.0.3_linux-x86-64.run',
},
Expand All @@ -22,7 +22,7 @@ const package_info = {
driver_version: '23.0.1',
firmware_version: '7.1.0.4.220',
firmware_url: 'https://ascend-repo.obs.cn-east-2.myhuaweicloud.com/Ascend HDK/Ascend HDK 23.0.2.1/Ascend-hdk-310p-npu-firmware_7.1.0.4.220.run',
kernel_url: 'https://ascend-repo.obs.cn-east-2.myhuaweicloud.com/Milan-ASL/Milan-ASL V100R001C18SPC702/Ascend-cann-kernels-310p_8.0.RC2.alpha002_linux.run',
kernel_url: 'https://ascend-repo.obs.cn-east-2.myhuaweicloud.com/Milan-ASL/Milan-ASL V100R001C19SPC802/Ascend-cann-kernels-310p_8.0.RC3.alpha001_linux.run',
x86_64: {
driver_url: 'https://ascend-repo.obs.cn-east-2.myhuaweicloud.com/Ascend HDK/Ascend HDK 23.0.2.1/Ascend-hdk-310p-npu-driver_23.0.1_linux-x86-64.run',
},
Expand All @@ -31,6 +31,38 @@ const package_info = {
}
}
},
'8.0.RC2': {
x86_64: {
url: 'https://ascend-repo.obs.cn-east-2.myhuaweicloud.com/CANN/CANN 8.0.RC2/Ascend-cann-toolkit_8.0.RC2_linux-x86_64.run'
},
aarch64: {
url: 'https://ascend-repo.obs.cn-east-2.myhuaweicloud.com/CANN/CANN 8.0.RC2/Ascend-cann-toolkit_8.0.RC2_linux-aarch64.run'
},
'910b': {
driver_version: '24.1.rc1',
firmware_version: '7.1.0.6.220',
firmware_url: 'https://ascend-repo.obs.cn-east-2.myhuaweicloud.com/Ascend HDK/Ascend HDK 24.1.RC1/Ascend-hdk-910b-npu-firmware_7.1.0.6.220.run',
kernel_url: 'https://ascend-repo.obs.cn-east-2.myhuaweicloud.com/CANN/CANN 8.0.RC2/Ascend-cann-kernels-910b_8.0.RC2_linux.run',
x86_64: {
driver_url: 'https://ascend-repo.obs.cn-east-2.myhuaweicloud.com/Ascend HDK/Ascend HDK 24.1.RC1/Ascend-hdk-910b-npu-driver_24.1.rc1_linux-x86-64.run',
},
aarch64: {
driver_url: 'https://ascend-repo.obs.cn-east-2.myhuaweicloud.com/Ascend HDK/Ascend HDK 24.1.RC1/Ascend-hdk-910b-npu-driver_24.1.rc1_linux-aarch64.run',
}
},
'310p': {
driver_version: '24.1.rc1',
firmware_version: '7.1.0.6.220',
firmware_url: 'https://ascend-repo.obs.cn-east-2.myhuaweicloud.com/Ascend HDK/Ascend HDK 24.1.RC1/Ascend-hdk-310p-npu-firmware_7.1.0.6.220.run',
kernel_url: 'https://ascend-repo.obs.cn-east-2.myhuaweicloud.com/CANN/CANN 8.0.RC2/Ascend-cann-kernels-310p_8.0.RC2_linux.run',
x86_64: {
driver_url: 'https://ascend-repo.obs.cn-east-2.myhuaweicloud.com/Ascend HDK/Ascend HDK 24.1.RC1/Ascend-hdk-310p-npu-driver_24.1.rc1_linux-x86-64.run',
},
aarch64: {
driver_url: 'https://ascend-repo.obs.cn-east-2.myhuaweicloud.com/Ascend HDK/Ascend HDK 24.1.RC1/Ascend-hdk-310p-npu-driver_24.1.rc1_linux-aarch64.run',
}
}
},
'8.0.RC1': {
x86_64: {
url: 'https://ascend-repo.obs.cn-east-2.myhuaweicloud.com/CANN/CANN 8.0.RC1/Ascend-cann-toolkit_8.0.RC1_linux-x86_64.run'
Expand Down Expand Up @@ -251,6 +283,12 @@ const docker_images = [
];

const pytorch_versions = {
'2.3.1': {
torch_npu: '2.3.1',
npu_branch: 'v2.3.1-6.0.rc2',
cann: 'CANN 8.0.RC2',
docker: 'cosdt/pytorch:2.3.1'
},
'2.2.0': {
torch_npu: '2.2.0',
npu_branch: 'v2.2.0-6.0.rc1',
Expand Down
2 changes: 2 additions & 0 deletions sources/pytorch/faq.rst
Original file line number Diff line number Diff line change
Expand Up @@ -4,6 +4,8 @@ FAQ
微信群
-----------

添加“开源小助手”微信,根据提示让小助手拉入群聊。

.. image:: ../../_static/images/pytorch_wechat.jpg
:width: 400px
:align: left
5 changes: 3 additions & 2 deletions sources/pytorch/install.rst
Original file line number Diff line number Diff line change
Expand Up @@ -23,7 +23,8 @@
<div class="row-element-2" id="col-values">
<div class="row" id="row-pytorch">
<div class="mobile-headings">PyTorch版本</div>
<div class="values-element block-3 install-pytorch selected" id="pytorch-2.2.0">2.2.0</div>
<div class="values-element block-3 install-pytorch selected" id="pytorch-2.3.1">2.3.1</div>
<div class="values-element block-3 install-pytorch" id="pytorch-2.2.0">2.2.0</div>
<div class="values-element block-3 install-pytorch" id="pytorch-2.1.0">2.1.0</div>
</div>
<div class="row" id="row-pytorch_npu">
Expand All @@ -50,7 +51,7 @@
</div>


2. 安装 PyTorch
1. 安装 PyTorch
----------------

.. warning::
Expand Down

0 comments on commit 525387a

Please sign in to comment.