diff --git a/Dockerfile b/Dockerfile
index 7352de7..194c5dd 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -3,13 +3,7 @@
 ###########
 
 # pull official base image
-FROM python:alpine as builder
-
-# update system
-RUN apk update && \
-    apk add git && \
-    apk add rsync && \
-    apk add docker
+FROM dvcorg/cml:0-dvc2-base1-gpu as builder
 
 # set work directory
 WORKDIR /usr/src/app
@@ -20,7 +14,7 @@ ENV PYTHONUNBUFFERED 1
 
 # install dependencies
 RUN pip install --upgrade pip
-COPY requirements.txt .
+COPY req-prod.txt requirements.txt
 RUN pip wheel --no-cache-dir --no-deps --wheel-dir /usr/src/app/wheels -r requirements.txt
 
 #########
@@ -28,7 +22,7 @@ RUN pip wheel --no-cache-dir --no-deps --wheel-dir /usr/src/app/wheels -r requir
 #########
 
 # pull official base image
-FROM python:alpine
+FROM dvcorg/cml:0-dvc2-base1-gpu
 
 # create app home
 ENV APP_HOME = /home/app
diff --git a/docker-compose.yml b/docker-compose.yml
new file mode 100644
index 0000000..d381f76
--- /dev/null
+++ b/docker-compose.yml
@@ -0,0 +1,7 @@
+version: "3.9"
+services:
+  gitea-runner:
+    build: .
+    container_name: gitea-runner
+    restart: unless-stopped
+    env_file: prod.env
\ No newline at end of file