From 3320482e12a566a3ac356b04bf9798e825797269 Mon Sep 17 00:00:00 2001
From: hotwa <pylyzeng@gmail.com>
Date: Sat, 15 Jun 2024 17:53:52 +0800
Subject: [PATCH] add

---
 finetune/docker-compose.yml | 32 ++++++++++++++++++++++++++++++++
 1 file changed, 32 insertions(+)
 create mode 100644 finetune/docker-compose.yml

diff --git a/finetune/docker-compose.yml b/finetune/docker-compose.yml
new file mode 100644
index 0000000..ebcf535
--- /dev/null
+++ b/finetune/docker-compose.yml
@@ -0,0 +1,32 @@
+version: '3.8'
+
+services:
+  ubuntu-finetune:
+    build: 
+      context: .
+      dockerfile: Dockerfile
+    volumes:
+      - ./src:/bbtft
+    container_name: ubuntu-finetune
+    pull_policy: if_not_present
+    tty: true
+    restart: unless-stopped
+    image: hotwa/finetune:test
+    ports:
+      - 3227:22
+    environment:
+      - NVIDIA_VISIBLE_DEVICES=all
+      - NVIDIA_DRIVER_CAPABILITIES=compute,utility
+    networks:
+      - network_finetune
+    deploy:
+      resources:
+        reservations:
+          devices:
+            - driver: nvidia
+              count: all
+              capabilities: [gpu]
+
+networks:
+  network_finetune:
+    name: network_finetune