diff --git a/Dockerfile b/Dockerfile
index 4c87cf9..b14e3ce 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,5 +1,5 @@
-# Python 3.11 + PyTorch 2.4.0 + CUDA 12.1 + CuDNN 9.5
-FROM pytorch/pytorch:2.4.0-cuda12.1-cudnn9-devel
+# Python 3.11 + PyTorch 2.7.0 + CUDA 12.8 + CuDNN 9.5
+FROM pytorch/pytorch:2.7.0-cuda12.8-cudnn9-devel
 
 # 기본 작업 경로 설정
 WORKDIR /app
diff --git a/docker-build.cmd b/docker-build.cmd
index 8a5a905..d932bff 100644
--- a/docker-build.cmd
+++ b/docker-build.cmd
@@ -1,3 +1,3 @@
-docker build --no-cache -t aicompanion/sdxl_train_captioner:0.9.3.
+docker build --no-cache -t aicompanion/sdxl_train_captioner:0.9.5 .
 
-docker tag aicompanion/sdxl_train_captioner:0.9.3 aicompanion/sdxl_train_captioner:latest
\ No newline at end of file
+docker tag aicompanion/sdxl_train_captioner:0.9.5 aicompanion/sdxl_train_captioner:latest
\ No newline at end of file
diff --git a/requirements.txt b/requirements.txt
index 5aab427..b8dba16 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -47,6 +47,6 @@ open-clip-torch==2.20.0
 rich==13.7.0
 
 # Kohya_ss library
-xformers==0.0.27.post2 --index-url https://download.pytorch.org/whl/cu121
-triton==2.3.0
+xformers --index-url https://download.pytorch.org/whl/cu128
+triton==3.0.0