diff --git a/.ci/docker/requirements.txt b/.ci/docker/requirements.txt
index b246193ff0..9109b6578d 100644
--- a/.ci/docker/requirements.txt
+++ b/.ci/docker/requirements.txt
@@ -49,6 +49,7 @@ onnxscript>=0.2.2
 onnxruntime
 evaluate
 accelerate>=0.20.1
+triton==3.4.0
 
 importlib-metadata==6.8.0