fix compose file

ai-dock · Dec 11, 2023 · 1de4d6f · 1de4d6f
1 parent e14bc00
commit 1de4d6f
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/docker-compose.yaml b/docker-compose.yaml
@@ -5,11 +5,11 @@ services:
     build:
       context: ./build
       args:
-        IMAGE_BASE: ${IMAGE_BASE:-ghcr.io/ai-dock/pytorch:2.1.0-py3.10-cuda-11.8.0-base-22.04}
+        IMAGE_BASE: ${IMAGE_BASE:-ghcr.io/ai-dock/pytorch:2.1.1-py3.10-cuda-11.8.0-base-22.04}
       tags:
-        - "ghcr.io/ai-dock/jupyter-pytorch:${IMAGE_TAG:-2.1.0-py3.10-cuda-11.8.0-base-22.04}"
+        - "ghcr.io/ai-dock/jupyter-pytorch:${IMAGE_TAG:-2.1.1-py3.10-cuda-11.8.0-base-22.04}"
 
-    image: ghcr.io/ai-dock/jupyter-pytorch:${IMAGE_TAG:-2.1.0-py3.10-cuda-11.8.0-base-22.04}
+    image: ghcr.io/ai-dock/jupyter-pytorch:${IMAGE_TAG:-2.1.1-py3.10-cuda-11.8.0-base-22.04}
 
     ## For Nvidia GPU's - You probably want to uncomment this
     #deploy: