diff --git a/docker-compose.simple.yml b/docker-compose.simple.yml
index 98e11b9..cbc0d29 100644
--- a/docker-compose.simple.yml
+++ b/docker-compose.simple.yml
@@ -3,9 +3,9 @@ version: '3.8'
 services:
   antsk:
     container_name: antsk
-    image: registry.cn-hangzhou.aliyuncs.com/xuzeyu91/antsk:v0.5.0 
+    image: registry.cn-hangzhou.aliyuncs.com/xuzeyu91/antsk:v0.5.1 
     # 如果需要pytorch环境需要使用下面这个镜像，镜像比较大
-    # image: registry.cn-hangzhou.aliyuncs.com/xuzeyu91/antsk:p0.5.0 
+    # image: registry.cn-hangzhou.aliyuncs.com/xuzeyu91/antsk:p0.5.1
     ports:
       - 5000:5000
     networks: