Skip to content

Commit

Permalink
update config and dockerfile
Browse files Browse the repository at this point in the history
  • Loading branch information
almazgimaev committed Dec 2, 2024
1 parent 74f7e1d commit 56ce83b
Show file tree
Hide file tree
Showing 4 changed files with 7 additions and 8 deletions.
3 changes: 1 addition & 2 deletions docker/Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -17,13 +17,12 @@ RUN pip3 install requests==2.28.2 urllib3==1.26.15
RUN pip3 install packaging==21.3
RUN pip3 install dill==0.3.8
RUN pip3 install ruamel.yaml==0.17.21
RUN python3 -m pip install pycocotools scikit-learn plotly==5.22.0

RUN pip3 install ultralytics==8.3.15
RUN pip3 install onnxruntime-gpu==1.18.0
RUN pip3 install dill==0.2.8.2 tf2onnx==1.16.1 skl2onnx==1.17.0
RUN pip3 uninstall -y nvidia-cudnn-cu11
RUN python3 -m pip install supervisely[tracking]==6.73.243
RUN python3 -m pip install supervisely[tracking,model-benchmark]==6.73.243
RUN python3 -c "from supervisely.nn.tracker import *"

LABEL python_sdk_version=6.73.243
4 changes: 2 additions & 2 deletions export_weights/config.json
Original file line number Diff line number Diff line change
Expand Up @@ -15,8 +15,8 @@
"files_file"
]
},
"docker_image": "supervisely/yolov8:1.0.38",
"instance_version": "6.12.5",
"docker_image": "supervisely/yolov8:1.0.39",
"instance_version": "6.12.12",
"main_script": "export_weights/src/main.py",
"modal_template": "export_weights/src/modal.html",
"modal_template_state": {
Expand Down
4 changes: 2 additions & 2 deletions serve/config.json
Original file line number Diff line number Diff line change
Expand Up @@ -20,8 +20,8 @@
"deployed_nn"
],
"community_agent": false,
"docker_image": "supervisely/yolov8:1.0.38",
"instance_version": "6.12.5",
"docker_image": "supervisely/yolov8:1.0.39",
"instance_version": "6.12.12",
"entrypoint": "python3 -m uvicorn src.main:m.app --app-dir ./serve --host 0.0.0.0 --port 8000 --ws websockets",
"port": 8000,
"task_location": "application_sessions",
Expand Down
4 changes: 2 additions & 2 deletions train/config.json
Original file line number Diff line number Diff line change
Expand Up @@ -11,8 +11,8 @@
"train"
],
"description": "Dashboard to configure, start and monitor YOLOv8 | v9 | v10 | v11 training",
"docker_image": "supervisely/yolov8:1.0.38",
"instance_version": "6.12.5",
"docker_image": "supervisely/yolov8:1.0.39",
"instance_version": "6.12.12",
"entrypoint": "python3 -m uvicorn src.main:app --app-dir ./train --host 0.0.0.0 --port 8000 --ws websockets",
"task_location": "workspace_tasks",
"need_gpu": true,
Expand Down

0 comments on commit 56ce83b

Please sign in to comment.