Create docker-compose-gpu.yml

docker-compose-gpu.yml

@@ -0,0 +1,12 @@
+version: '3'
+
+services:
+  flashlight_gpu:
+    image: flml/flashlight:cuda-latest
+    runtime: nvidia
+    tty: true
+    stdin_open: true
+    container_name: flashlight_gpu
+    volumes:
+      - ./data/:/data
+      - ./models/:/models