Use different tags for GPU versions of serverless functions (cvat-ai#…

…7215) Having both CPU and GPU images use the same image name and tag is confusing, since you can't tell which one you have on your system without deep inspection.
amjadsaadeh · Dec 14, 2023 · 3642590 · 3642590
1 parent 32f0ff7
commit 3642590
Show file tree

Hide file tree

Showing 9 changed files with 13 additions and 8 deletions.
diff --git a/changelog.d/20231204_124652_roman.md b/changelog.d/20231204_124652_roman.md
@@ -0,0 +1,5 @@
+### Changed
+
+- GPU versions of serverless functions now use the `latest-gpu` Docker tag
+  rather than `latest`
+  (<https://github.com/opencv/cvat/pull/7215>)
diff --git a/serverless/onnx/WongKinYiu/yolov7/nuclio/function-gpu.yaml b/serverless/onnx/WongKinYiu/yolov7/nuclio/function-gpu.yaml
@@ -95,7 +95,7 @@ spec:
   handler: main:handler
   eventTimeout: 30s
   build:
-    image: cvat.onnx.wongkinyiu.yolov7
+    image: cvat.onnx.wongkinyiu.yolov7:latest-gpu
     baseImage: nvidia/cuda:11.8.0-cudnn8-runtime-ubuntu22.04
 
     directives:

diff --git a/serverless/pytorch/dschoerk/transt/nuclio/function-gpu.yaml b/serverless/pytorch/dschoerk/transt/nuclio/function-gpu.yaml
@@ -17,7 +17,7 @@ spec:
       value: /opt/nuclio/trans-t
 
   build:
-    image: cvat.pth.dschoerk.transt
+    image: cvat.pth.dschoerk.transt:latest-gpu
     baseImage: nvidia/cuda:11.7.1-devel-ubuntu20.04
 
     directives:

diff --git a/serverless/pytorch/facebookresearch/detectron2/retinanet_r101/nuclio/function-gpu.yaml b/serverless/pytorch/facebookresearch/detectron2/retinanet_r101/nuclio/function-gpu.yaml
@@ -96,7 +96,7 @@ spec:
   eventTimeout: 30s
 
   build:
-    image: cvat.pth.facebookresearch.detectron2.retinanet_r101
+    image: cvat.pth.facebookresearch.detectron2.retinanet_r101:latest-gpu
     baseImage: ubuntu:20.04
 
     directives:

diff --git a/serverless/pytorch/facebookresearch/sam/nuclio/function-gpu.yaml b/serverless/pytorch/facebookresearch/sam/nuclio/function-gpu.yaml
@@ -26,7 +26,7 @@ spec:
       value: /opt/nuclio/sam
 
   build:
-    image: cvat.pth.facebookresearch.sam.vit_h
+    image: cvat.pth.facebookresearch.sam.vit_h:latest-gpu
     baseImage: ubuntu:22.04
 
     directives:

diff --git a/serverless/pytorch/foolwood/siammask/nuclio/function-gpu.yaml b/serverless/pytorch/foolwood/siammask/nuclio/function-gpu.yaml
@@ -17,7 +17,7 @@ spec:
       value: /opt/nuclio/SiamMask:/opt/nuclio/SiamMask/experiments/siammask_sharp
 
   build:
-    image: cvat.pth.foolwood.siammask
+    image: cvat.pth.foolwood.siammask:latest-gpu
     baseImage: nvidia/cuda:11.1.1-devel-ubuntu20.04
 
     directives:

diff --git a/serverless/pytorch/saic-vul/hrnet/nuclio/function-gpu.yaml b/serverless/pytorch/saic-vul/hrnet/nuclio/function-gpu.yaml
@@ -22,7 +22,7 @@ spec:
       value: /opt/nuclio/hrnet
 
   build:
-    image: cvat.pth.saic-vul.hrnet
+    image: cvat.pth.saic-vul.hrnet:latest-gpu
     baseImage: ubuntu:20.04
 
     directives:

diff --git a/serverless/tensorflow/faster_rcnn_inception_v2_coco/nuclio/function-gpu.yaml b/serverless/tensorflow/faster_rcnn_inception_v2_coco/nuclio/function-gpu.yaml
@@ -96,7 +96,7 @@ spec:
   eventTimeout: 30s
 
   build:
-    image: cvat.tf.faster_rcnn_inception_v2_coco
+    image: cvat.tf.faster_rcnn_inception_v2_coco:latest-gpu
     baseImage: tensorflow/tensorflow:2.1.1-gpu
 
     directives:

diff --git a/serverless/tensorflow/matterport/mask_rcnn/nuclio/function-gpu.yaml b/serverless/tensorflow/matterport/mask_rcnn/nuclio/function-gpu.yaml
@@ -100,7 +100,7 @@ spec:
     - name: MASK_RCNN_DIR
       value: /opt/nuclio/Mask_RCNN
   build:
-    image: cvat.tf.matterport.mask_rcnn
+    image: cvat.tf.matterport.mask_rcnn:latest-gpu
     baseImage: tensorflow/tensorflow:1.15.5-gpu-py3
     directives:
       postCopy: