diff --git a/Dockerfile b/Dockerfile
index 1894b6355..03dd3a855 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,16 +1,19 @@
 FROM python:3.10-slim
 
 
-ENV CLI_ARGS=""
-
-COPY . /opt/comfy_ui
 WORKDIR /opt/comfy_ui
 
+
 RUN PIP_NO_CACHE_DIR=1 pip install --extra-index-url https://download.pytorch.org/whl/cu117 \
-torch==1.13.1 \
+torch==1.13.1+cu117 \
 torchvision \
 torchaudio \
-xformers
+xformers \
+triton
+
+COPY . /opt/comfy_ui
 RUN pip install -r requirements.txt
 
+
+ENV CLI_ARGS=""
 CMD python main.py ${CLI_ARGS}