diff --git a/Dockerfile b/Dockerfile
index 6ef7070da..39df23381 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -15,7 +15,8 @@ RUN apt-get update && apt-get install -y build-essential
 
 # test
 RUN cat /proc/cpuinfo
-RUN cat /proc/cpuinfo | grep avx2
+RUN uv pip install --system py-cpuinfo
+RUN python test_cpu.py
 
 # 安装依赖
 RUN uv pip install --system --upgrade pip
diff --git a/test_cpu.py b/test_cpu.py
new file mode 100644
index 000000000..befb40ec5
--- /dev/null
+++ b/test_cpu.py
@@ -0,0 +1,5 @@
+import cpuinfo
+
+cpu_info = cpuinfo.get_cpu_info()
+print(f"当前cpu信息:{cpu_info}")
+print(f"当前cpu指令集支持：{cpu_info["flags"]}")
\ No newline at end of file