feat: 增加torch2.4.1的检查方法

dcu/pytorch_check.ipynb

@@ -9,21 +9,13 @@
     "此脚本为pytorch的DCU镜像通用检查脚本，更多项目和issue信息请参考[链接](https://developer.sourcefind.cn/codes/chenpangpang/gpu-base-image-test)。"
    ]
   },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "de0f0d91-c642-4a34-bb2a-1c7b482246c9",
-   "metadata": {},
-   "outputs": [],
-   "source": []
-  },
   {
    "cell_type": "markdown",
    "id": "3b01123f-d38f-4078-9e4c-4770ead409b0",
    "metadata": {},
    "source": [
     "## 一、线上版本检测\n",
-    "目前线上版本检测分为(1) torch 2.1.0 python 3.8 (2) torch 2.1.0 python 3.10 (3) python 2.3.0 三个版本的检测"
+    "目前线上版本检测分为(1) torch 2.1.0 python 3.8 (2) torch 2.1.0 python 3.10 (3) python 2.3.0 (4) python 2.4.1 四个版本的检测"
    ]
   },
   {
@@ -71,6 +63,16 @@
     "import faiss; print(faiss.__version__,'\\n',faiss.__dcu_version__)"
    ]
   },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "d68a5555-4175-43a4-8873-3818ab867371",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "import torch;print(torch.__version__)"
+   ]
+  },
   {
    "cell_type": "markdown",
    "id": "a49bb205-254c-433c-8fa8-a9fecfbe1127",
@@ -144,6 +146,31 @@
     "import torchvision;print(torchvision.__version__,'\\n',torchvision.__dcu_version__)"
    ]
   },
+  {
+   "cell_type": "markdown",
+   "id": "7f47ca42-4b83-4a1b-9136-488fd8a56135",
+   "metadata": {},
+   "source": [
+    "## 1.4 torch 2.4.1 检测"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "ab8de4e9-f46e-4ec9-8b78-17f1bbb17ad3",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "import os\n",
+    "os.system(\"cat /etc/issue\")\n",
+    "import torch\n",
+    "print(torch.cuda.is_available())\n",
+    "import sys\n",
+    "print(sys.path)\n",
+    "import torch;print(torch.__version__,'\\n',torch.__hcu_version__)\n",
+    "import torchvision;print(torchvision.__version__,'\\n',torchvision.__hcu_version__)"
+   ]
+  },
   {
    "cell_type": "markdown",
    "id": "7d6b5669-7323-439a-81a9-fd4d22203374",
@@ -223,7 +250,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.8.15"
+   "version": "3.10.18"
   }
  },
  "nbformat": 4,