Fix cuda version (#4)

* remove nvidia-cudnn-cu12 * remove nvidia-cudnn-cu12 * Update release_branch.yml * update docker * update docker * update docker * update config * update config * update docker * update docker
supervisely-ecosystem · Oct 25, 2024 · f21cb5f · f21cb5f
1 parent 08c0059
commit f21cb5f
Show file tree

Hide file tree

Showing 4 changed files with 13 additions and 16 deletions.
diff --git a/docker/Dockerfile b/docker/Dockerfile
@@ -1,10 +1,11 @@
-FROM supervisely/base-py-sdk:6.73.216
+FROM nvidia/cuda:11.7.1-cudnn8-devel-ubuntu22.04
 ENV DEBIAN_FRONTEND=noninteractive
-RUN apt-get update
+RUN apt update && apt install python3-pip -y
+RUN apt-get install -y git
 RUN apt-get install -y curl
-RUN apt install -y nvidia-cuda-toolkit
+RUN apt-get install ffmpeg libgeos-dev libsm6 libxext6 libexiv2-dev libxrender-dev libboost-all-dev -y
 
-RUN pip3 install -U nvidia-cudnn-cu12==8.9.7.29
+RUN pip3 install networkx==2.8.8
 RUN pip3 install torch==2.0.1 torchvision==0.15.2 --index-url https://download.pytorch.org/whl/cu117
 RUN pip3 uninstall onnxruntime -y
 RUN pip3 install onnx==1.14.0 onnxruntime-gpu==1.15.1
@@ -27,6 +28,4 @@ RUN curl -L -o /app/repo/models/rtdetr_r50vd_coco_objects365.pth https://github.
 RUN curl -L -o /app/repo/models/rtdetr_r101vd_coco_objects365.pth https://github.com/lyuwenyu/storage/releases/download/v0.1/rtdetr_r101vd_2x_coco_objects365_from_paddle.pth
 
 RUN python3 -m pip install supervisely==6.73.216
-RUN python3 -m pip install supervisely[model-benchmark]==6.73.216
-
-LABEL python_sdk_version=6.73.216
+RUN python3 -m pip install supervisely[model-benchmark]==6.73.216
diff --git a/docker/publish.sh b/docker/publish.sh
@@ -1,2 +1,2 @@
-docker build -t supervisely/rt-detr:1.0.3 . && \
-docker push supervisely/rt-detr:1.0.3
+docker build -t supervisely/rt-detr:1.0.4 . && \
+docker push supervisely/rt-detr:1.0.4
diff --git a/supervisely_integration/serve/config.json b/supervisely_integration/serve/config.json
@@ -11,9 +11,8 @@
 	"gpu": "preferred",
 	"session_tags": ["deployed_nn"],
 	"community_agent": false,
-	"docker_image": "supervisely/rt-detr:1.0.3",
-	"instance_version": "6.8.88",
-	"entrypoint": "python -m uvicorn main:model.app --host 0.0.0.0 --port 8000 --ws websockets --app-dir supervisely_integration/serve",
+	"docker_image": "supervisely/rt-detr:1.0.4",
+	"entrypoint": "python3 -m uvicorn main:model.app --host 0.0.0.0 --port 8000 --ws websockets --app-dir supervisely_integration/serve",
 	"port": 8000,
 	"task_location": "application_sessions",
 	"license": {

diff --git a/supervisely_integration/train/config.json b/supervisely_integration/train/config.json
@@ -2,14 +2,14 @@
 	"type": "app",
 	"version": "2.0.0",
 	"name": "Train RT-DETR",
-	"entrypoint": "python -m uvicorn main:app --host 0.0.0.0 --port 8000 --ws websockets --app-dir supervisely_integration/train",
+	"entrypoint": "python3 -m uvicorn main:app --host 0.0.0.0 --port 8000 --ws websockets --app-dir supervisely_integration/train",
 	"port": 8000,
 	"categories": ["neural network", "images", "object detection", "train"],
 	"icon": "https://github.com/user-attachments/assets/b4f91538-7f2b-46f5-baf9-16527fa7c1a7",
 	"icon_cover": true,
 	"poster": "https://github.com/user-attachments/assets/ae6ec55e-63eb-43d1-99de-f62650939c69",
 	"description": "Train RT-DETR model on your data",
-	"docker_image": "supervisely/rt-detr:1.0.3",
+	"docker_image": "supervisely/rt-detr:1.0.4",
 	"instance_version": "6.11.22",
 	"task_location": "workspace_tasks",
 	"need_gpu": true,
@@ -21,6 +21,5 @@
 	},
 	"license": {
 		"type": "Apache-2.0"
-	},
-	"instance_version": "6.11.22"
+	}
 }