Deploy: updated MedCAT service env vars and container defs for GPU se…

…tups.
CogStack · Feb 20, 2024 · 1b16638 · 1b16638
1 parent bc8fd35
commit 1b16638
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 4 deletions.
diff --git a/deploy/nlp_service.env b/deploy/nlp_service.env
@@ -5,4 +5,13 @@ RES_BIOYODIE_UMLS_PATH="../services/nlp-services/applications/bio-yodie/resource
 # For other NLP vars and settings please view the following files
 # MEDCAT_SERVICE:
 #   - ../services/nlp-services/applications/medcat/config/env_app
-#   - ../services/nlp-services/applications/medcat/config/env_medcat
+#   - ../services/nlp-services/applications/medcat/config/env_medcat
+
+
+# MedCAT sergvice image versions:
+# default image, only CPU support: cogstacksystems/medcat-service:latest 
+# GPU support: cogstacksystems/medcat-service-gpu:latest
+MEDCAT_SERVICE_IMAGE_VERSION="cogstacksystems/medcat-service:0.6.0"
+
+# ENABLE THIS ONLY IF YOU ARE USING GPU
+APP_CUDA_DEVICE_COUNT=0
diff --git a/deploy/services.yml b/deploy/services.yml
@@ -253,15 +253,13 @@ services:
 #   * using internal models / resources                                     #
 #---------------------------------------------------------------------------#
   nlp-medcat-service-production:
-    image: cogstacksystems/medcat-service:0.6.0
+    image: ${MEDCAT_SERVICE_IMAGE_VERSION:-cogstacksystems/medcat-service:0.6.0}
     container_name: cogstack-medcat-service-production
     restart: always
     env_file:
       - ./nlp_service.env
       - ../services/nlp-services/applications/medcat/config/env_app
       - ../services/nlp-services/applications/medcat/config/env_medcat
-    environment:
-      - RES_MEDCAT_SERVICE_MODEL_PRODUCTION_PATH=${RES_MEDCAT_SERVICE_MODEL_PRODUCTION_PATH:-../services/nlp-services/applications/medcat/models/}
     volumes:
       - ./${RES_MEDCAT_SERVICE_MODEL_PRODUCTION_PATH:-../services/nlp-services/applications/medcat/models/}:/cat/models/:rw
     expose: