add manual TCP port mapping

ai-dock · Nov 22, 2023 · 916fdce · 916fdce
1 parent 754976d
commit 916fdce
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/build/COPY_ROOT/opt/ai-dock/fastapi/serviceportal/helpers.py b/build/COPY_ROOT/opt/ai-dock/fastapi/serviceportal/helpers.py
@@ -48,6 +48,7 @@ def get_cfqt_url(port):
         return False
 
 def get_direct_url(port):
+    port = os.environ.get("MAPPED_TCP_PORT_" + port, port)
     direct_address = os.environ.get('DIRECT_ADDRESS')
     if direct_address == 'auto#vast-ai':
         return get_vast_url(port)

diff --git a/docker-compose.yaml b/docker-compose.yaml
@@ -15,9 +15,9 @@ services:
         ROCM_VERSION: ${ROCM_VERSION:-5.6}
         ROCM_LEVEL: ${ROCM_LEVEL:-runtime}
       tags:
-        - "ghcr.io/ai-dock/base-image:${IMAGE_TAG}"
+        - "ghcr.io/ai-dock/base-image:${IMAGE_TAG:-cuda-11.8.0-base-22.04}"
 
-    image: ghcr.io/ai-dock/base-image:${IMAGE_TAG}
+    image: ghcr.io/ai-dock/base-image:${IMAGE_TAG:-cuda-11.8.0-base-22.04}
 
     ## For Nvidia GPU's - You probably want to uncomment this
     #deploy: