update versions

Ascend · Jun 24, 2024 · 63e37db · 63e37db
1 parent 0d5abab
commit 63e37db
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 19 deletions.
diff --git a/_static/package_info.js b/_static/package_info.js
@@ -177,18 +177,18 @@ const package_info = {
 
 
 const docker_images = [
-    'cosdt/cann:7.0.0-310p-ubuntu20.04',
-    'cosdt/cann:7.0.0-310p-ubuntu22.04',
-    'cosdt/cann:7.0.0-310p-openeuler20.03',
-    'cosdt/cann:7.0.0-910b-ubuntu20.04',
-    'cosdt/cann:7.0.0-910b-ubuntu22.04',
-    'cosdt/cann:7.0.0-910b-openeuler20.03',
-    'cosdt/cann:8.0.rc1-310p-ubuntu20.04',
-    'cosdt/cann:8.0.rc1-310p-ubuntu22.04',
-    'cosdt/cann:8.0.rc1-310p-openeuler20.03',
-    'cosdt/cann:8.0.rc1-910b-ubuntu20.04',
-    'cosdt/cann:8.0.rc1-910b-ubuntu22.04',
-    'cosdt/cann:8.0.rc1-910b-openeuler20.03',
+    'cosdt/cann:7.0.1-310p-ubuntu20.04',
+    'cosdt/cann:7.0.1-310p-ubuntu22.04',
+    'cosdt/cann:7.0.1-310p-openeuler20.03',
+    'cosdt/cann:7.0.1-910b-ubuntu20.04',
+    'cosdt/cann:7.0.1-910b-ubuntu22.04',
+    'cosdt/cann:7.0.1-910b-openeuler20.03',
+    'cosdt/cann:8.0.rc2.alpha002-310p-ubuntu20.04',
+    'cosdt/cann:8.0.rc2.alpha002-310p-ubuntu22.04',
+    'cosdt/cann:8.0.rc2.alpha002-310p-openeuler20.03',
+    'cosdt/cann:8.0.rc2.alpha002-910b-ubuntu20.04',
+    'cosdt/cann:8.0.rc2.alpha002-910b-ubuntu22.04',
+    'cosdt/cann:8.0.rc2.alpha002-910b-openeuler20.03',
 ];
 
 const pytorch_versions = {
@@ -203,11 +203,5 @@ const pytorch_versions = {
         npu_branch: 'v2.1.0-6.0.rc1',
         cann: 'CANN 8.0.RC1',
         docker: 'cosdt/pytorch:2.1.0'
-    },
-    '2.0.1': {
-        torch_npu: '2.0.1.post1',
-        npu_branch: 'v2.1.0-5.0.0',
-        cann: 'CANN 7.0.0',
-        docker: 'cosdt/pytorch:2.0.1'
     }
 }
diff --git a/_static/pytorch_actions.js b/_static/pytorch_actions.js
@@ -57,8 +57,9 @@ docker run \\
     -v /usr/local/dcmi:/usr/local/dcmi \\
     -v /usr/local/bin/npu-smi:/usr/local/bin/npu-smi \\
     -v /usr/local/Ascend/driver/lib64/:/usr/local/Ascend/driver/lib64/ \\
-    -v /etc/ascend_install.info:/etc/ascend_install.info \\
     -v /usr/local/Ascend/driver/version.info:/usr/local/Ascend/driver/version.info \\
+    -v /etc/ascend_install.info:/etc/ascend_install.info \\
+    -e DRIVER_PATH=/usr/local/Ascend/driver \\
     -it ${match_versions['docker']} bash
              `;