diff --git a/Dockerfile b/Dockerfile
index d13bae9..4b6055e 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -7,7 +7,7 @@ WORKDIR /app
 # 필수 패키지 설치
 RUN sed -i 's|archive.ubuntu.com|mirror.kakao.com|g' /etc/apt/sources.list && \
     apt-get update && \
-    apt-get install -y apt-utils && \
+    apt-get install -y apt-utils libgl1 libglib2.0-0 && \
     apt-get install -y --no-install-recommends git wget curl && \
     rm -rf /var/lib/apt/lists/*
 
diff --git a/docker-compose.yml b/docker-compose.yml
index b331a0b..60193be 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -5,6 +5,7 @@ services:
       dockerfile: Dockerfile
     image: aicompanion/sdxl_train_captioner:latest
     container_name: sdxl_train_captioner
+    runtime: nvidia
 
     # GPU 설정
     deploy:
@@ -15,9 +16,6 @@ services:
               count: all
               capabilities: [gpu]
 
-    # ports:
-    #   - "7860:7860"
-
     # 볼륨 마운트 (호스트 ↔ 컨테이너)
     volumes:
       - ./models:/app/sdxl_train_captioner/models
@@ -27,7 +25,8 @@ services:
 
     environment:
       # GPU 선택 (필요 시 GPU ID 지정)
-      - CUDA_VISIBLE_DEVICES=3
+      - NVIDIA_VISIBLE_DEVICES=all      
+      - CUDA_VISIBLE_DEVICES=0
       - HF_HOME=/app/sdxl_train_captioner/models
       - HF_HUB_CACHE=/app/sdxl_train_captioner/models
       - PYTHONUNBUFFERED=1
diff --git a/docker-up.cmd b/docker-up.cmd
index 5177d11..a8d0f87 100644
--- a/docker-up.cmd
+++ b/docker-up.cmd
@@ -1 +1 @@
-docker-compose up -d
\ No newline at end of file
+docker-compose up -d --gpus all
\ No newline at end of file