Skip to content

Commit

Permalink
update docker files
Browse files Browse the repository at this point in the history
  • Loading branch information
Borg93 committed May 16, 2024
1 parent b5d3373 commit 6c9a252
Show file tree
Hide file tree
Showing 5 changed files with 251 additions and 247 deletions.
50 changes: 0 additions & 50 deletions Dockerfile

This file was deleted.

File renamed without changes.
14 changes: 14 additions & 0 deletions docker/Dockerfile
Original file line number Diff line number Diff line change
@@ -0,0 +1,14 @@
FROM borg93/cuda-12-py310:0.0.1
WORKDIR /app
COPY ./dist/htrflow_core-0.0.1-py3-none-any.whl /tmp/
RUN python3 -m pip install /tmp/htrflow_core-0.0.1-py3-none-any.whl[local_models,cli] \
&& rm /tmp/htrflow_core-0.0.1-py3-none-any.whl \
&& apt-get clean \
&& rm -rf /var/lib/apt/lists/* \
&& cd /app \
&& mkdir outputs input pipeline logs models
# CMD ["htrflow", "--help"]
# docker build -t borg93/htrflow_im:0.0.1 .
# docker push borg93/htrflow_im:0.0.1
# docker run --gpus all -it --rm htrflow_im:latest /bin/bash
# docker exec -it <c_id> /bin/bash
36 changes: 36 additions & 0 deletions docker/docker-compose.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,36 @@
version: "3.8"

services:
htrflow:
image: htrflow-test

deploy:
resources:
reservations:
devices:
- driver: nvidia
count: 1
capabilities: [gpu]

command:
[
"/bin/sh",
"-c",
"htrflow pipeline pipeline/im-pipeline.yaml input/1107 --logfile logs/htrflow/htrflow_core_1107.log",
]

volumes:
- ./input-volume:/app/input
- ./output-volume:/app/outputs
- ./logs-volume:/app/logs
- ./pipeline:/app/pipeline
- ./.cache:/app/models

configs:
config.yaml:
external: false
file: im-pipeline.yaml

networks:
default:
driver: bridge
Loading

0 comments on commit 6c9a252

Please sign in to comment.