diff --git a/docker-compose.simple.yml b/docker-compose.simple.yml
index cb9e810..b06aeb4 100644
--- a/docker-compose.simple.yml
+++ b/docker-compose.simple.yml
@@ -3,7 +3,9 @@ version: '3.8'
 services:
   antsk:
     container_name: antsk
-    image: registry.cn-hangzhou.aliyuncs.com/xuzeyu91/antsk:v0.2.1
+    image: registry.cn-hangzhou.aliyuncs.com/xuzeyu91/antsk:v0.2.3
+    # 如果需要pytorch环境需要使用下面这个镜像，镜像比较大
+    # image: registry.cn-hangzhou.aliyuncs.com/xuzeyu91/antsk:v0.2.3.2
     ports:
       - 5000:5000
     networks:
diff --git a/docker-compose.yml b/docker-compose.yml
index 536794b..fac0c3e 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -19,6 +19,8 @@ services:
   antsk:
     container_name: antsk
     image: registry.cn-hangzhou.aliyuncs.com/xuzeyu91/antsk:v0.2.3
+    # 如果需要pytorch环境需要使用下面这个镜像，镜像比较大
+    # image: registry.cn-hangzhou.aliyuncs.com/xuzeyu91/antsk:v0.2.3.2
     ports:
       - 5000:5000
     networks: