diff --git a/docker-compose-cuda.yml b/docker-compose-cuda.yml
new file mode 100644
index 0000000..3b99827
--- /dev/null
+++ b/docker-compose-cuda.yml
@@ -0,0 +1,17 @@
+version: '3.8'
+services:
+  alltalk-tts:
+    image: joshxt/alltalk_tts:cuda
+    restart: unless-stopped
+    ports:
+      - "7851:7851"
+    volumes:
+      - ./outputs:/app/outputs/
+      - ./finetune/put-voice-samples-in-here:/app/finetune/put-voice-samples-in-here
+    deploy:
+      resources:
+        reservations:
+          devices:
+            - driver: nvidia
+              count: all
+              capabilities: [ gpu ]
\ No newline at end of file
diff --git a/docker-compose.yml b/docker-compose.yml
new file mode 100644
index 0000000..d54c3ee
--- /dev/null
+++ b/docker-compose.yml
@@ -0,0 +1,10 @@
+version: '3.8'
+services:
+  alltalk-tts:
+    image: joshxt/alltalk_tts:cpu
+    restart: unless-stopped
+    ports:
+      - "7851:7851"
+    volumes:
+      - ./outputs:/app/outputs/
+      - ./finetune/put-voice-samples-in-here:/app/finetune/put-voice-samples-in-here