11const package_info = {
2+ '8.2.RC1' : {
3+ x86_64 : {
4+ url : 'https://ascend-repo.obs.cn-east-2.myhuaweicloud.com/CANN/CANN 8.2.RC1/Ascend-cann-toolkit_8.2.RC1_linux-x86_64.run' ,
5+ } ,
6+ aarch64 : {
7+ url : 'https://ascend-repo.obs.cn-east-2.myhuaweicloud.com/CANN/CANN 8.2.RC1/Ascend-cann-toolkit_8.2.RC1_linux-aarch64.run' ,
8+ } ,
9+ '910b' : {
10+ driver_version : '25.0.rc1' ,
11+ firmware_version : '7.7.0.1.231' ,
12+ firmware_url : 'https://ascend-repo.obs.cn-east-2.myhuaweicloud.com/Ascend HDK/Ascend HDK 25.0.RC1.1/Ascend-hdk-910b-npu-firmware_7.7.0.1.231.run' ,
13+ x86_64 : {
14+ driver_url : 'https://ascend-repo.obs.cn-east-2.myhuaweicloud.com/Ascend HDK/Ascend HDK 25.0.RC1.1/Ascend-hdk-910b-npu-driver_25.0.rc1.1_linux-x86-64.run' ,
15+ kernel_url : 'https://ascend-repo.obs.cn-east-2.myhuaweicloud.com/CANN/CANN 8.2.RC1/Ascend-cann-kernels-910b_8.2.RC1_linux-x86_64.run' ,
16+ } ,
17+ aarch64 : {
18+ driver_url : 'https://ascend-repo.obs.cn-east-2.myhuaweicloud.com/Ascend HDK/Ascend HDK 25.0.RC1.1/Ascend-hdk-910b-npu-driver_25.0.rc1.1_linux-aarch64.run' ,
19+ kernel_url : 'https://ascend-repo.obs.cn-east-2.myhuaweicloud.com/CANN/CANN 8.2.RC1/Ascend-cann-kernels-910b_8.2.RC1_linux-aarch64.run' ,
20+ }
21+ } ,
22+ '310p' : {
23+ driver_version : '25.0.rc1' ,
24+ firmware_version : '7.7.0.1.231' ,
25+ firmware_url : 'https://ascend-repo.obs.cn-east-2.myhuaweicloud.com/Ascend HDK/Ascend HDK 25.0.RC1.1/Ascend-hdk-310p-npu-firmware_7.7.0.1.231.run' ,
26+ x86_64 : {
27+ driver_url : 'https://ascend-repo.obs.cn-east-2.myhuaweicloud.com/Ascend HDK/Ascend HDK 25.0.RC1.1/Ascend-hdk-310p-npu-driver_25.0.rc1.1_linux-x86-64.run' ,
28+ kernel_url : 'https://ascend-repo.obs.cn-east-2.myhuaweicloud.com/CANN/CANN 8.2.RC1/Ascend-cann-kernels-310p_8.2.RC1_linux-x86_64.run' ,
29+ } ,
30+ aarch64 : {
31+ driver_url : 'https://ascend-repo.obs.cn-east-2.myhuaweicloud.com/Ascend HDK/Ascend HDK 25.0.RC1.1/Ascend-hdk-310p-npu-driver_25.0.rc1.1_linux-aarch64.run' ,
32+ kernel_url : 'https://ascend-repo.obs.cn-east-2.myhuaweicloud.com/CANN/CANN 8.2.RC1/Ascend-cann-kernels-310p_8.2.RC1_linux-aarch64.run' ,
33+ }
34+ }
35+ } ,
236 '8.1.RC1' : {
337 x86_64 : {
4- url : 'https://ascend-repo.obs.cn-east-2.myhuaweicloud.com/Milan-ASL/Milan-ASL V100R001C21B800TP034 /Ascend-cann-toolkit_8.1.RC1.alpha001_linux -x86_64.run' ,
38+ url : 'https://ascend-repo.obs.cn-east-2.myhuaweicloud.com/CANN/CANN 8.1.RC1 /Ascend-cann-toolkit_8.1.RC1_linux -x86_64.run' ,
539 } ,
640 aarch64 : {
7- url : 'https://ascend-repo.obs.cn-east-2.myhuaweicloud.com/Milan-ASL/Milan-ASL V100R001C21B800TP034 /Ascend-cann-toolkit_8.1.RC1.alpha001_linux -aarch64.run' ,
41+ url : 'https://ascend-repo.obs.cn-east-2.myhuaweicloud.com/CANN/CANN 8.1.RC1 /Ascend-cann-toolkit_8.1.RC1_linux -aarch64.run' ,
842 } ,
943 '910b' : {
1044 driver_version : '24.1.rc2' ,
1145 firmware_version : '7.3.0.1.231' ,
1246 firmware_url : 'https://ascend-repo.obs.cn-east-2.myhuaweicloud.com/Ascend HDK/Ascend HDK 24.1.RC2/Ascend-hdk-910b-npu-firmware_7.3.0.1.231.run' ,
1347 x86_64 : {
14- driver_url : 'https://ascend-repo.obs.cn-east-2.myhuaweicloud.com/Ascend HDK/Ascend HDK 24.1.RC2/Ascend-hdk-910b-npu-driver_24.1.rc2_linux-x86_64 .run' ,
15- kernel_url : 'https://ascend-repo.obs.cn-east-2.myhuaweicloud.com/Milan-ASL/Milan-ASL V100R001C21B800TP034 /Ascend-cann-kernels-910b_8.1.RC1.alpha001_linux -x86_64.run' ,
48+ driver_url : 'https://ascend-repo.obs.cn-east-2.myhuaweicloud.com/Ascend HDK/Ascend HDK 24.1.RC2/Ascend-hdk-910b-npu-driver_24.1.rc2_linux-x86-64 .run' ,
49+ kernel_url : 'https://ascend-repo.obs.cn-east-2.myhuaweicloud.com/CANN/CANN 8.1.RC1 /Ascend-cann-kernels-910b_8.1.RC1_linux -x86_64.run' ,
1650 } ,
1751 aarch64 : {
1852 driver_url : 'https://ascend-repo.obs.cn-east-2.myhuaweicloud.com/Ascend HDK/Ascend HDK 24.1.RC2/Ascend-hdk-910b-npu-driver_24.1.rc2_linux-aarch64.run' ,
19- kernel_url : 'https://ascend-repo.obs.cn-east-2.myhuaweicloud.com/Milan-ASL/Milan-ASL V100R001C21B800TP034 /Ascend-cann-kernels-910b_8.1.RC1.alpha001_linux -aarch64.run' ,
53+ kernel_url : 'https://ascend-repo.obs.cn-east-2.myhuaweicloud.com/CANN/CANN 8.1.RC1 /Ascend-cann-kernels-910b_8.1.RC1_linux -aarch64.run' ,
2054 }
2155 } ,
2256 '310p' : {
@@ -25,11 +59,11 @@ const package_info = {
2559 firmware_url : 'https://ascend-repo.obs.cn-east-2.myhuaweicloud.com/Ascend HDK/Ascend HDK 24.1.RC2/Ascend-hdk-310p-npu-firmware_7.3.0.1.231.run' ,
2660 x86_64 : {
2761 driver_url : 'https://ascend-repo.obs.cn-east-2.myhuaweicloud.com/Ascend HDK/Ascend HDK 24.1.RC2/Ascend-hdk-310p-npu-driver_24.1.rc2_linux-x86-64.run' ,
28- kernel_url : 'https://ascend-repo.obs.cn-east-2.myhuaweicloud.com/Milan-ASL/Milan-ASL V100R001C21B800TP034 /Ascend-cann-kernels-310p_8.1.RC1.alpha001_linux -x86_64.run' ,
62+ kernel_url : 'https://ascend-repo.obs.cn-east-2.myhuaweicloud.com/CANN/CANN 8.1.RC1 /Ascend-cann-kernels-310p_8.1.RC1_linux -x86_64.run' ,
2963 } ,
3064 aarch64 : {
3165 driver_url : 'https://ascend-repo.obs.cn-east-2.myhuaweicloud.com/Ascend HDK/Ascend HDK 24.1.RC2/Ascend-hdk-310p-npu-driver_24.1.rc2_linux-aarch64.run' ,
32- kernel_url : 'https://ascend-repo.obs.cn-east-2.myhuaweicloud.com/Milan-ASL/Milan-ASL V100R001C21B800TP034 /Ascend-cann-kernels-310p_8.1.RC1.alpha001_linux -aarch64.run' ,
66+ kernel_url : 'https://ascend-repo.obs.cn-east-2.myhuaweicloud.com/CANN/CANN 8.1.RC1 /Ascend-cann-kernels-310p_8.1.RC1_linux -aarch64.run' ,
3367 }
3468 }
3569 } ,
@@ -45,7 +79,7 @@ const package_info = {
4579 firmware_version : '7.3.0.1.231' ,
4680 firmware_url : 'https://ascend-repo.obs.cn-east-2.myhuaweicloud.com/Ascend HDK/Ascend HDK 24.1.RC2/Ascend-hdk-910b-npu-firmware_7.3.0.1.231.run' ,
4781 x86_64 : {
48- driver_url : 'https://ascend-repo.obs.cn-east-2.myhuaweicloud.com/Ascend HDK/Ascend HDK 24.1.RC2/Ascend-hdk-910b-npu-driver_24.1.rc2_linux-x86_64 .run' ,
82+ driver_url : 'https://ascend-repo.obs.cn-east-2.myhuaweicloud.com/Ascend HDK/Ascend HDK 24.1.RC2/Ascend-hdk-910b-npu-driver_24.1.rc2_linux-x86-64 .run' ,
4983 kernel_url : 'https://ascend-repo.obs.cn-east-2.myhuaweicloud.com/CANN/CANN 8.0.0/Ascend-cann-kernels-910b_8.0.0_linux-x86_64.run' ,
5084 } ,
5185 aarch64 : {
@@ -79,7 +113,7 @@ const package_info = {
79113 firmware_version : '7.3.0.1.231' ,
80114 firmware_url : 'https://ascend-repo.obs.cn-east-2.myhuaweicloud.com/Ascend HDK/Ascend HDK 24.1.RC2/Ascend-hdk-910b-npu-firmware_7.3.0.1.231.run' ,
81115 x86_64 : {
82- driver_url : 'https://ascend-repo.obs.cn-east-2.myhuaweicloud.com/Ascend HDK/Ascend HDK 24.1.RC2/Ascend-hdk-910b-npu-driver_24.1.rc2_linux-x86_64 .run' ,
116+ driver_url : 'https://ascend-repo.obs.cn-east-2.myhuaweicloud.com/Ascend HDK/Ascend HDK 24.1.RC2/Ascend-hdk-910b-npu-driver_24.1.rc2_linux-x86-64 .run' ,
83117 kernel_url : 'https://ascend-repo.obs.cn-east-2.myhuaweicloud.com/CANN/CANN 8.0.RC3/Ascend-cann-kernels-910b_8.0.RC3_linux-x86_64.run' ,
84118 } ,
85119 aarch64 : {
@@ -345,28 +379,28 @@ const docker_images = [
345379
346380// Only show 3 latest versions
347381const pytorch_versions = {
348- '2.4.0 ' : {
349- torch_npu : '2.4.0.post2 ' ,
350- npu_branch : 'v2.4.0 -6.0.0' ,
351- cann : 'CANN 8.0.0 ' ,
352- docker : 'ascendai/pytorch:2.4.0 '
382+ '2.7.1 ' : {
383+ torch_npu : '2.7.1.post6 ' ,
384+ npu_branch : 'v2.7.1 -6.0.0' ,
385+ cann : 'CANN 8.2.RC1 ' ,
386+ docker : 'ascendai/pytorch:2.7.1 '
353387 } ,
354- '2.3.1 ' : {
355- torch_npu : '2.3.1.post4 ' ,
356- npu_branch : 'v2.3.1- 6.0.0' ,
357- cann : 'CANN 8.0.0 ' ,
358- docker : 'ascendai/pytorch:2.3.1 '
388+ '2.6.0 ' : {
389+ torch_npu : '2.6.0 ' ,
390+ npu_branch : 'v2.6.0-7.1 .0' ,
391+ cann : 'CANN 8.2.RC1 ' ,
392+ docker : 'ascendai/pytorch:2.6.0 '
359393 } ,
360- '2.2.0 ' : {
361- torch_npu : '2.2.0.post2 ' ,
362- npu_branch : 'v2.2 .0-6.0.rc2 ' ,
363- cann : 'CANN 8.0.RC2 ' ,
364- docker : 'ascendai/pytorch:2.2.0 '
394+ '2.5.1 ' : {
395+ torch_npu : '2.5.1.post1 ' ,
396+ npu_branch : 'v2.6 .0-7.1.0 ' ,
397+ cann : 'CANN 8.2.RC1 ' ,
398+ docker : 'ascendai/pytorch:2.5.1 '
365399 } ,
366- '2.1 .0' : {
367- torch_npu : '2.1 .0.post10 ' ,
368- npu_branch : 'v2.1 .0-6.0.0' ,
369- cann : 'CANN 8.0.0 ' ,
370- docker : 'ascendai/pytorch:2.1 .0'
400+ '2.4 .0' : {
401+ torch_npu : '2.4 .0.post4 ' ,
402+ npu_branch : 'v2.4 .0-6.0.0' ,
403+ cann : 'CANN 8.1.RC1 ' ,
404+ docker : 'ascendai/pytorch:2.4 .0'
371405 }
372406}
0 commit comments