diff --git a/requirements-cuda.txt b/requirements-cuda.txt index 10596ed85d60..c8b58258b0f6 100644 --- a/requirements-cuda.txt +++ b/requirements-cuda.txt @@ -5,6 +5,8 @@ ray >= 2.9 nvidia-ml-py # for pynvml package torch == 2.3.0 +triton >= 3.0.0 + # These must be updated alongside torch torchvision == 0.18.0 # Required for phi3v processor, also see https://github.com/pytorch/vision?tab=readme-ov-file#installation for corresponding version xformers == 0.0.26.post1 # Requires PyTorch 2.3.0