test: test查看cpu指令集

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

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"]}")