Update Dockerfile

Dockerfile

@@ -1,5 +1,6 @@
 
-FROM nvidia/cuda:12.1.0-cudnn8-runtime-ubuntu22.04
+# FROM nvidia/cuda:12.1.0-cudnn8-runtime-ubuntu22.04
+FROM nvidia/cuda:12.9.1-cudnn-runtime-ubuntu24.04
 
 ENV VLLM_LOGGING_LEVEL=DEBUG
 ENV HF_HOME=/tmp/.cache/huggingface
@@ -181,7 +182,7 @@ RUN uv pip install --system --index-strategy unsafe-best-match vllm==0.10.0 --ex
 # INFO:     Waiting for application shutdown.
 # INFO:     Application shutdown complete.
 # INFO:     Finished server process [27]
-RUN uv pip install --system --index-strategy unsafe-best-match triton==3.2 --extra-index-url https://download.pytorch.org/whl/cu128
+# RUN uv pip install --system --index-strategy unsafe-best-match triton==3.2 --extra-index-url https://download.pytorch.org/whl/cu128
 
 # # Then, install xformers with the --no-build-isolation flag
 # RUN uv pip install --system \